Create scripts/download_ckpts.py
Browse files
scripts/download_ckpts.py
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from huggingface_hub import snapshot_download
|
| 2 |
+
|
| 3 |
+
# Download only best_model.pth and config.json to a local directory
|
| 4 |
+
snapshot_download(
|
| 5 |
+
repo_id="danhtran2mind/Viet-Glow-TTS-finetuning",
|
| 6 |
+
local_dir="./ckpts",
|
| 7 |
+
local_dir_use_symlinks=False,
|
| 8 |
+
allow_patterns=["best_model.pth", "config.json"]
|
| 9 |
+
)
|