cosrigel commited on
Commit
f43b7b0
·
verified ·
1 Parent(s): 4395db8

Update app_local.py

Browse files
Files changed (1) hide show
  1. app_local.py +3 -2
app_local.py CHANGED
@@ -18,9 +18,10 @@ import os
18
  from huggingface_hub import hf_hub_download, snapshot_download
19
  #sys.path.append("dia-finetuning/dia")
20
 
21
- TMP_DIR = Path(os.getenv("TMP_DIR", "/data/tmp"))
22
  TMP_DIR.mkdir(parents=True, exist_ok=True)
23
 
 
24
  DAC_MODEL = None # cache DAC để không tải lại mỗi lần reload_model
25
 
26
  # --- Resolve checkpoints for Space ---
@@ -32,7 +33,7 @@ def _resolve_checkpoint():
32
  repo = os.getenv("MODEL_REPO") # ví dụ: "cosrigel/dia-finetuning-vnese"
33
  model_file = os.getenv("MODEL_FILE") # ví dụ: "model.safetensors" hoặc "model.pth"
34
  override_dir = os.getenv("CKPT_DIR") # nếu muốn trỏ thư mục local
35
- out_dir = os.getenv("HF_WEIGHTS_DIR", "/data/dia_ckpt")
36
  Path(out_dir).mkdir(parents=True, exist_ok=True)
37
 
38
  # 1) CKPT_DIR (ưu tiên nếu được set)
 
18
  from huggingface_hub import hf_hub_download, snapshot_download
19
  #sys.path.append("dia-finetuning/dia")
20
 
21
+ TMP_DIR = Path(os.getenv("TMP_DIR", "/data/tmp").strip())
22
  TMP_DIR.mkdir(parents=True, exist_ok=True)
23
 
24
+
25
  DAC_MODEL = None # cache DAC để không tải lại mỗi lần reload_model
26
 
27
  # --- Resolve checkpoints for Space ---
 
33
  repo = os.getenv("MODEL_REPO") # ví dụ: "cosrigel/dia-finetuning-vnese"
34
  model_file = os.getenv("MODEL_FILE") # ví dụ: "model.safetensors" hoặc "model.pth"
35
  override_dir = os.getenv("CKPT_DIR") # nếu muốn trỏ thư mục local
36
+ out_dir = os.getenv("HF_WEIGHTS_DIR", "/data/dia_ckpt").strip()
37
  Path(out_dir).mkdir(parents=True, exist_ok=True)
38
 
39
  # 1) CKPT_DIR (ưu tiên nếu được set)