Update app.py

app.py

@@ -4,7 +4,7 @@ import random
 import spaces
 import torch
 from diffusers import  DiffusionPipeline, FlowMatchEulerDiscreteScheduler, AutoencoderTiny, AutoencoderKL
-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 transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
 from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
 import multiprocessing as mp
@@ -50,10 +50,10 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
     else:
         joint_attention_kwargs = None
 
-    cache_scope = apply_cache_on_pipe(
-        pipe,
-        # residual_diff_threshold=0.2,
-    )
+    # apply_cache_on_pipe(
+    #     pipe,
+    #     # residual_diff_threshold=0.2,
+    # )
     
     try:
         # Call the custom pipeline function with the correct keyword argument
@@ -67,7 +67,6 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
             output_type="pil",
             good_vae=good_vae,  # Assuming good_vae is defined elsewhere
             joint_attention_kwargs=joint_attention_kwargs,  # Fixed parameter name
-            cache_scope=cache_scope,
         ):
             yield img, seed
     finally: