PreethiCarmelBosco commited on
Commit
236a3c7
·
verified ·
1 Parent(s): 6c7a3a5

resolve error

Browse files
Files changed (1) hide show
  1. 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.")