Update wan2_fast.py

wan2_fast.py

@@ -8,6 +8,9 @@ from huggingface_hub import hf_hub_download
 import numpy as np
 import random
 import os
+
+
+
 # LIGHT WEIGHT 1.3b
 # MODEL_ID = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
 # LORA_REPO_ID = "Kijai/WanVideo_comfy"
@@ -16,7 +19,7 @@ import os
 
 ## This is working well ##
 
-#MODEL_ID = "FastVideo/FastWan2.2-TI2V-5B-FullAttn-Diffusers"
+MODEL_ID = "FastVideo/FastWan2.2-TI2V-5B-FullAttn-Diffusers"
 
 ## below model has a blury output but it get loaded 
 ##MODEL_ID ="Runware/Wan2.2-TI2V-5B"
@@ -26,7 +29,7 @@ import os
 #MODEL_ID ="Wan-AI/Wan2.2-T2V-A14B-Diffusers"
 
 
-MODEL_ID ="linoyts/Wan2.2-T2V-A14B-Diffusers-BF16"
+#MODEL_ID ="linoyts/Wan2.2-T2V-A14B-Diffusers-BF16"
 
 
 
@@ -51,6 +54,18 @@ torch.cuda.empty_cache()
 
 # hold lora for now
 
+LORA_REPO_ID = "rahul7star/wan2.2Lora"
+LORA_FILENAME = "wan2.2/Wan2.2-Lightning_I2V-A14B-4steps-lora_LOW_fp16.safetensors"
+causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
+pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
+pipe.set_adapters(["causvid_lora"], adapter_weights=[0.95])
+pipe.fuse_lora()
+
+
+
+
+
+
 # LORA_REPO_ID = "AlekseyCalvin/Phenakistiscopes_Wan22_5B_T2V_LoRA"
 # LORA_FILENAME = "PhenakistiscopesV4_000002300.safetensors"
 # causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)