Spaces:
Running
Running
Update app_local.py
Browse files- 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)
|