Update wan2_fast.py

wan2_fast.py

@@ -16,7 +16,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,13 +26,13 @@ MODEL_ID = "FastVideo/FastWan2.2-TI2V-5B-FullAttn-Diffusers"
 #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"
 
 
 
 
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 
 
@@ -47,16 +47,16 @@ pipe = WanPipeline.from_pretrained(
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=8.0)
 pipe.to("cuda")
 
-
+torch.cuda.empty_cache()
 
 # hold lora for now
 
-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)
-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)
+# 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"