Update app.py

app.py

@@ -11,8 +11,8 @@ from diffusers.models.attention_processor import AttnProcessor2_0
 from custom_pipeline import FluxWithCFGPipeline
 
 # --- Torch Optimizations ---
-# torch.backends.cuda.matmul.allow_tf32 = True
-# torch.backends.cudnn.benchmark = True # Enable cuDNN benchmark for potentially faster convolutions
+torch.backends.cuda.matmul.allow_tf32 = True
+torch.backends.cudnn.benchmark = True # Enable cuDNN benchmark for potentially faster convolutions
 
 # --- Constants ---
 MAX_SEED = np.iinfo(np.int32).max
@@ -36,7 +36,6 @@ pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtyp
 pipe.to(device)
 
 # Apply optimizations
-pipe.unet.set_attn_processor(AttnProcessor2_0())
 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")