Dan Flower commited on
Commit
a5b1d48
·
1 Parent(s): c8a274c

Fix filename case and add debug to download_model.py

Browse files
Files changed (1) hide show
  1. model/download_model.py +26 -15
model/download_model.py CHANGED
@@ -2,20 +2,31 @@
2
  import os, pathlib, sys
3
  from huggingface_hub import hf_hub_download
4
 
5
- MODEL_REPO = os.getenv("MODEL_REPO", "TheBloke/TinyLlama-1.1B-Chat-v1.0-GGUF")
6
- MODEL_FILE = os.getenv("MODEL_FILE", "TinyLlama-1.1B-Chat-v1.0.Q4_K_M.gguf")
7
- MODEL_DIR = pathlib.Path(os.getenv("MODEL_DIR", "/tmp/models"))
8
- MODEL_DIR.mkdir(parents=True, exist_ok=True)
 
9
 
10
- print(f"[download] repo={MODEL_REPO} file={MODEL_FILE} -> {MODEL_DIR}", file=sys.stderr)
11
- path = hf_hub_download(
12
- repo_id=MODEL_REPO,
13
- filename=MODEL_FILE,
14
- local_dir=str(MODEL_DIR),
15
- local_dir_use_symlinks=False,
16
- token=os.getenv("HF_TOKEN") or None,
17
- )
18
- print(f"[download] saved: {path}", file=sys.stderr)
 
19
 
20
- # Optional: write MODEL_PATH file for debugging
21
- print(path, file=open(MODEL_DIR / "MODEL_PATH.txt", "w"))
 
 
 
 
 
 
 
 
 
 
2
  import os, pathlib, sys
3
  from huggingface_hub import hf_hub_download
4
 
5
+ def main():
6
+ MODEL_REPO = os.getenv("MODEL_REPO", "TheBloke/TinyLlama-1.1B-Chat-v1.0-GGUF")
7
+ MODEL_FILE = os.getenv("MODEL_FILE", "tinyllama-1.1b-chat-v1.0.Q4_K_M.gguf") # <- fixed case
8
+ MODEL_DIR = pathlib.Path(os.getenv("MODEL_DIR", "/tmp/models"))
9
+ MODEL_DIR.mkdir(parents=True, exist_ok=True)
10
 
11
+ try:
12
+ print(f"[download] repo={MODEL_REPO} file={MODEL_FILE} -> {MODEL_DIR}", file=sys.stderr, flush=True)
13
+ path = hf_hub_download(
14
+ repo_id=MODEL_REPO,
15
+ filename=MODEL_FILE,
16
+ local_dir=str(MODEL_DIR),
17
+ local_dir_use_symlinks=False,
18
+ token=os.getenv("HF_TOKEN") or None,
19
+ )
20
+ print(f"[download] saved: {path}", file=sys.stderr, flush=True)
21
 
22
+ # Optional debug file
23
+ with open(MODEL_DIR / "MODEL_PATH.txt", "w") as f:
24
+ f.write(path + "\n")
25
+
26
+ except Exception as e:
27
+ print(f"!!! Exception in download_model.py: {e}", file=sys.stderr, flush=True)
28
+ import traceback
29
+ traceback.print_exc()
30
+
31
+ if __name__ == "__main__":
32
+ main()