new lora path

app.py

@@ -275,15 +275,16 @@ def ui(GPU_memory_mode, scheduler_dict, config_path, compile_dit, weight_dtype):
             
             # Use snapshot download for the VideoCoF repo to get all weights (including safetensors)
             try:
-                from huggingface_hub import snapshot_download
+                from huggingface_hub import snapshot_download, hf_hub_download
                 print("Downloading Wan2.1-T2V-14B weights...")
                 snapshot_download(repo_id="Wan-AI/Wan2.1-T2V-14B", local_dir="Wan-AI/Wan2.1-T2V-14B")
                 print("Downloading VideoCoF weights...")
-                snapshot_download(repo_id="XiangpengYang/VideoCoF")
+                os.makedirs("models/Personalized_Model", exist_ok=True)
+                hf_hub_download(repo_id="XiangpengYang/VideoCoF", filename="videocof.safetensors", local_dir="models/Personalized_Model")
             except Exception as e:
                 print(f"Warning: Failed to pre-download weights: {e}")
 
-            base_model_dropdown, lora_model_dropdown, lora_alpha_slider, _ = create_finetune_models_checkpoints(controller, visible=False, default_lora="XiangpengYang/VideoCoF")
+            base_model_dropdown, lora_model_dropdown, lora_alpha_slider, _ = create_finetune_models_checkpoints(controller, visible=False, default_lora="videocof.safetensors")
             
             # Set default LoRA alpha to 1.0 (matching inference.py)
             lora_alpha_slider.value = 1.0