Update app.py

app.py

@@ -33,7 +33,7 @@ import shutil
 import yaml
 import numpy as np
 
-from huggingface_hub import hf_hub_download
+from huggingface_hub import snapshot_download
 
 print(os.getcwd())
 os.environ["TORCH_HOME"] = os.path.join(os.getcwd(), "checkpoints")
@@ -57,8 +57,7 @@ config = get_train_config(config_path)
 model.config = config
 store_attn_map = False
 
-file_path = hf_hub_download(repo_id="ByteDance/XVerse", force_download=False)
-ckpt_root = os.path.dirname(file_path)
+ckpt_root = snapshot_download(repo_id="ByteDance/XVerse")
 
 modulation_adapter = load_modulation_adapter(model, config, dtype, device, f"{ckpt_root}/modulation_adapter", is_training=False)
 model.add_modulation_adapter(modulation_adapter)