Update wan2_fast.py

wan2_fast.py

@@ -19,17 +19,9 @@ MODEL_ID = "FastVideo/FastWan2.2-TI2V-5B-FullAttn-Diffusers"
 #MODEL_ID ="FastWan/FastWan_T2V_14B_480p_lora_rank_128_bf16.safetensors"
 
 
-LORA_REPO_ID = "Kijai/WanVideo_comfy"
-LORA_FILENAME = "Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank256_bf16.safetensors"
-#LORA_FILENAME = "Pusa/Wan21_PusaV1_LoRA_14B_rank512_bf16.safetensors"
-
 
 
 
-# LORA_REPO_ID = "RaphaelLiu/PusaV1"
-# LORA_FILENAME="pusa_v1.safetensors"
-#LORA_REPO_ID = "Kijai/WanVideo_comfy"
-#LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
 
 
 
@@ -45,10 +37,17 @@ pipe = WanPipeline.from_pretrained(
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=8.0)
 pipe.to("cuda")
 
-# 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 = "Kijai/WanVideo_comfy"
+LORA_FILENAME = "Wan22-Lightning/Wan2.2-Lightning_T2V-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()
 
 # MOD_VALUE = 32
 # DEFAULT_H_SLIDER_VALUE =  512