fix

infer.py

@@ -123,6 +123,13 @@ def parse_args():
 def load_models(args):
     # Load Wan I2V models
     model_manager = ModelManager(device="cpu")
+    ckpt_path = hf_hub_download(configs[name].repo_id, configs[name].repo_flow.replace("sft", "safetensors"))
+    
+    if hf_download:
+        # lora_ckpt_path = hf_hub_download("bytedance-research/UNO", "dit_lora.safetensors")
+        try:
+            lora_ckpt_path = hf_hub_download("bytedance-research/UNO", "dit_lora.safetensors")
+            
     model_manager.load_models(
         [
             [