Update model_manager.py

model_manager.py

@@ -215,18 +215,31 @@ class ModelManager:
         loop = asyncio.get_running_loop()
 
         def _dl(repo, glob):
-            import subprocess, shutil
+            import subprocess
+            # Build env — HF_TOKEN optional for public repos
+            env = {**os.environ, "HF_HUB_ENABLE_HF_TRANSFER": "1"}
+            token = os.getenv("HF_TOKEN", "").strip()
+            if token:
+                env["HUGGING_FACE_HUB_TOKEN"] = token
+                env["HF_TOKEN"] = token
             for g in [glob, m.glob_fallback if glob == m.glob_primary else ""]:
                 if not g:
                     continue
-                r = subprocess.run(
-                    ["huggingface-cli", "download", repo,
-                     "--include", g, "--local-dir", str(MODEL_DIR), "--quiet"],
-                    env={**os.environ, "HF_HUB_ENABLE_HF_TRANSFER": "1"},
-                    capture_output=True
-                )
+                cmd = ["huggingface-cli", "download", repo,
+                       "--include", g, "--local-dir", str(MODEL_DIR)]
+                if not token:
+                    # Public access — verbose to see any auth errors
+                    cmd.append("--quiet")
+                else:
+                    cmd.append("--quiet")
+                logger.info(f"[models] Running: {' '.join(cmd[:5])}...")
+                r = subprocess.run(cmd, env=env, capture_output=True, timeout=600)
                 if r.returncode == 0:
-                    break
+                    logger.info(f"[models] Download succeeded for glob: {g}")
+                    return ""
+                else:
+                    err = r.stderr.decode("utf-8", errors="replace")[:300]
+                    logger.warning(f"[models] Download failed (glob={g}): {err}")
             return ""
 
         TIMEOUT = float(os.getenv("DOWNLOAD_TIMEOUT", "30.0")) * 20  # max 10 min
@@ -429,16 +442,24 @@ if __name__ == "__main__":
 
     elif cmd == "ensure-downloaded":
         async def _run():
+            token = os.getenv("HF_TOKEN", "").strip()
+            print(f"[model_manager] HF_TOKEN: {'SET (len=' + str(len(token)) + ')' if token else 'NOT SET — using anonymous access'}", file=sys.stderr)
             mgr.apply_pending()
             m = CATALOG_BY_ID.get(mgr.cfg.active_id) or mgr.select_best()
             mgr.cfg.active_id = m.id
+            print(f"[model_manager] Downloading: {m.name} ({m.size_gb}GB) from {m.repo}", file=sys.stderr)
             path, mmproj = await mgr.download(m)
             if not path:
-                # fallback to smallest
+                print(f"[model_manager] Primary download failed, trying smallest fallback...", file=sys.stderr)
                 fb = min(CATALOG, key=lambda x: x.size_gb)
+                print(f"[model_manager] Fallback: {fb.name}", file=sys.stderr)
                 path, mmproj = await mgr.download(fb)
                 if path:
                     mgr.cfg.active_id = fb.id
+            if not path:
+                print(f"[model_manager] ALL downloads failed!", file=sys.stderr)
+            else:
+                print(f"[model_manager] SUCCESS: {path}", file=sys.stderr)
             mgr.cfg.active_path = path
             mgr.cfg.mmproj_path = mmproj
             mgr.cfg.save()
@@ -451,4 +472,4 @@ if __name__ == "__main__":
         asyncio.run(_run())
 
     else:
-        print(f"Unknown: {cmd}", file=sys.stderr); sys.exit(1)
+        print(f"Unknown: {cmd}", file=sys.stderr); sys.exit(1)