Spaces:
Build error
Build error
resolve error
Browse files- download_model.py +2 -3
download_model.py
CHANGED
|
@@ -4,15 +4,14 @@ import os
|
|
| 4 |
# This script runs ONCE during the Docker build
|
| 5 |
print("Starting model download...")
|
| 6 |
|
|
|
|
| 7 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 8 |
|
| 9 |
hf_hub_download(
|
| 10 |
repo_id="mradermacher/prem-1B-SQL-GGUF",
|
| 11 |
filename="prem-1B-SQL.Q8_0.gguf",
|
| 12 |
token=HF_TOKEN,
|
| 13 |
-
local_dir="."
|
| 14 |
-
local_dir_use_symlinks=False,
|
| 15 |
-
local_path_and_filename="prem-1B-SQL.Q8_0.gguf"
|
| 16 |
)
|
| 17 |
|
| 18 |
print("Model download complete.")
|
|
|
|
| 4 |
# This script runs ONCE during the Docker build
|
| 5 |
print("Starting model download...")
|
| 6 |
|
| 7 |
+
# Get the token from the build secret
|
| 8 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 9 |
|
| 10 |
hf_hub_download(
|
| 11 |
repo_id="mradermacher/prem-1B-SQL-GGUF",
|
| 12 |
filename="prem-1B-SQL.Q8_0.gguf",
|
| 13 |
token=HF_TOKEN,
|
| 14 |
+
local_dir="." # This will save the file as ./prem-1B-SQL.Q8_0.gguf
|
|
|
|
|
|
|
| 15 |
)
|
| 16 |
|
| 17 |
print("Model download complete.")
|