Spaces:
Paused
Paused
Update services/vince_server.py
Browse files- services/vince_server.py +7 -5
services/vince_server.py
CHANGED
|
@@ -33,7 +33,9 @@ class VinceServer:
|
|
| 33 |
self.config_path = config_path
|
| 34 |
self.output_root = Path(output_root); self.output_root.mkdir(parents=True, exist_ok=True)
|
| 35 |
overrides = list(base_overrides or [])
|
| 36 |
-
|
|
|
|
|
|
|
| 37 |
# 1) Clonar/garantir repo
|
| 38 |
self.ensure_repo()
|
| 39 |
|
|
@@ -105,11 +107,11 @@ class VinceServer:
|
|
| 105 |
repo_id=VINCE_REPO_ID,
|
| 106 |
filename=rel_path,
|
| 107 |
local_dir=str(VINCE_CKPT),
|
| 108 |
-
|
|
|
|
| 109 |
token=token,
|
| 110 |
-
force_download=False,
|
| 111 |
-
local_files_only=False,
|
| 112 |
-
revision=repo_revision,
|
| 113 |
)
|
| 114 |
downloaded += 1
|
| 115 |
except Exception as de:
|
|
|
|
| 33 |
self.config_path = config_path
|
| 34 |
self.output_root = Path(output_root); self.output_root.mkdir(parents=True, exist_ok=True)
|
| 35 |
overrides = list(base_overrides or [])
|
| 36 |
+
self.HF_HOME = Path(os.getenv("HF_HOME", "/data/.cache/huggingface"))
|
| 37 |
+
self.HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN") or None
|
| 38 |
+
|
| 39 |
# 1) Clonar/garantir repo
|
| 40 |
self.ensure_repo()
|
| 41 |
|
|
|
|
| 107 |
repo_id=VINCE_REPO_ID,
|
| 108 |
filename=rel_path,
|
| 109 |
local_dir=str(VINCE_CKPT),
|
| 110 |
+
cache_dir=str(self.HF_HOME),
|
| 111 |
+
#local_dir_use_symlinks=False,
|
| 112 |
token=token,
|
| 113 |
+
#force_download=False,
|
| 114 |
+
#local_files_only=False,
|
|
|
|
| 115 |
)
|
| 116 |
downloaded += 1
|
| 117 |
except Exception as de:
|