danhtran2mind commited on
Commit
e8ed146
·
verified ·
1 Parent(s): d13455f

Create scripts/download_ckpts.py

Browse files
Files changed (1) hide show
  1. scripts/download_ckpts.py +9 -0
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
+ )