Update app.py

app.py

@@ -33,6 +33,8 @@ import shutil
 import yaml
 import numpy as np
 
+from huggingface_hub import hf_hub_download
+
 dtype = torch.bfloat16
 device = "cuda"
 
@@ -52,7 +54,9 @@ config = get_train_config(config_path)
 model.config = config
 store_attn_map = False
 
-ckpt_root = "~/.cache/huggingface/hub/XVerse"
+file_path = hf_hub_download(repo_id="ByteDance/XVerse", force_download=False)
+ckpt_root = os.path.dirname(file_path)
+
 modulation_adapter = load_modulation_adapter(model, config, dtype, device, f"{ckpt_root}/modulation_adapter", is_training=False)
 model.add_modulation_adapter(modulation_adapter)
 if config["model"]["use_dit_lora"]: