Update app_fast.py

app_fast.py

@@ -18,28 +18,20 @@ vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=to
 text_to_video_pipe = WanPipeline.from_pretrained(MODEL_ID, vae=vae, torch_dtype=torch.bfloat16)
 image_to_video_pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID, vae=vae, torch_dtype=torch.bfloat16)
 
+#apply_first_block_cache(pipe.transformer, FirstBlockCacheConfig(threshold=0.2))
+
 for pipe in [text_to_video_pipe, image_to_video_pipe]:
     pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
-    pipe.to("cuda")
-
-apply_first_block_cache(pipe.transformer, FirstBlockCacheConfig(threshold=0.2))
-
-
+    pipe.to("cuda")    
 ##Lora testing
 # LORA_REPO_ID = "JERRYNPC/WAN2.2-LORA-NSFW"
 # LORA_FILENAME= "jerry_HIGH-nsfw-V10E800.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/HSToric_Color_Wan2.2_5B_LoRA_BySilverAgePoets"
 #LORA_FILENAME = "HSToric_color_Wan22_5b_LoRA.safetensors"