Update app.py

app.py

@@ -2,7 +2,7 @@ import torch
 from diffusers import UniPCMultistepScheduler
 from diffusers import WanPipeline, AutoencoderKLWan
 from diffusers.models.transformers.transformer_wan import WanTransformer3DModel
-from para_attn.first_block_cache.diffusers_adapters import apply_cache_on_pipe
+# from para_attn.first_block_cache.diffusers_adapters import apply_cache_on_pipe
 from huggingface_hub import hf_hub_download
 from PIL import Image
 import numpy as np
@@ -150,7 +150,7 @@ def generate(prompt, negative_prompt, width, height, num_inference_steps, option
     print(f"Setting weights for this run: {list(zip(ALL_ADAPTER_NAMES, adapter_weights))}")
     pipe.set_adapters(ALL_ADAPTER_NAMES, adapter_weights=adapter_weights)
 
-    apply_cache_on_pipe(pipe)
+    # apply_cache_on_pipe(pipe)
     
     # --- Step 3: Run inference ---
     output = pipe(