euIaxs22 commited on
Commit
bc73c13
·
verified ·
1 Parent(s): 41f3876

Update services/vince_server.py

Browse files
Files changed (1) hide show
  1. 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
- local_dir_use_symlinks=False,
 
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: