Dan Flower commited on
Commit
7880051
·
1 Parent(s): f9dc060

troubleshooting model path issue

Browse files
Files changed (1) hide show
  1. model/download_model.py +15 -8
model/download_model.py CHANGED
@@ -1,27 +1,34 @@
1
- #!/usr/bin/env python3
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)
 
1
+ import os, sys, pathlib, shutil
 
2
  from huggingface_hub import hf_hub_download
3
 
4
  def main():
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_PATH = MODEL_DIR / MODEL_FILE
9
+
10
  MODEL_DIR.mkdir(parents=True, exist_ok=True)
11
 
12
  try:
13
+ print(f"[download] repo={MODEL_REPO} file={MODEL_FILE} -> {MODEL_PATH}", file=sys.stderr, flush=True)
14
+ # Download (to cache, possibly symlinked)
15
+ downloaded = hf_hub_download(
16
  repo_id=MODEL_REPO,
17
  filename=MODEL_FILE,
18
  local_dir=str(MODEL_DIR),
19
  local_dir_use_symlinks=False,
20
  token=os.getenv("HF_TOKEN") or None,
21
  )
22
+ print(f"[download] hf_hub_download saved: {downloaded}", file=sys.stderr, flush=True)
23
+
24
+ # Copy to expected flat path if different
25
+ if pathlib.Path(downloaded).resolve() != MODEL_PATH.resolve():
26
+ shutil.copy2(downloaded, MODEL_PATH)
27
+ print(f"[download] copied to: {MODEL_PATH}", file=sys.stderr, flush=True)
28
 
29
+ # Write debug marker
30
  with open(MODEL_DIR / "MODEL_PATH.txt", "w") as f:
31
+ f.write(str(MODEL_PATH) + "\n")
32
 
33
  except Exception as e:
34
  print(f"!!! Exception in download_model.py: {e}", file=sys.stderr, flush=True)