Update app.py

app.py

@@ -5,9 +5,6 @@ import spaces
 import torch
 import time
 from diffusers import DiffusionPipeline, AutoencoderTiny
-# Using AttnProcessor2_0 for potential speedup with PyTorch 2.x
-from diffusers.models.attention_processor import AttnProcessor2_0
-# Assuming custom_pipeline defines FluxWithCFGPipeline correctly
 from custom_pipeline import FluxWithCFGPipeline
 
 # --- Torch Optimizations ---
@@ -36,8 +33,6 @@ pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtyp
 pipe.to(device)
 
 # Apply optimizations
-pipe.vae.set_attn_processor(AttnProcessor2_0()) # VAE might benefit too
-
 pipe.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v2.3.safetensors', adapter_name="better")
 pipe.set_adapters(["better"], adapter_weights=[1.0])
 pipe.fuse_lora(adapter_name=["better"], lora_scale=1.0) # Fuse for potential speedup