Update app.py

app.py

@@ -45,10 +45,6 @@ from PIL import Image
 from image_gen_aux import UpscaleWithModel
 
 
-from diffusers.models.attention_processor import AttnProcessor2_0
-from diffusers.models.attention_processor import Attention
-        
-
 # --- GCS Configuration ---
 # Make sure to set these secrets in your Hugging Face Space settings
 GCS_BUCKET_NAME = os.getenv("GCS_BUCKET_NAME")
@@ -105,13 +101,8 @@ def load_model():
     pipe.transformer=ll_transformer
     pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
     pipe.to(device=device, dtype=torch.bfloat16)
-    #for name, module in pipe.transformer.named_modules():
-    #    if isinstance(module, Attention):
-    #        module.processor = fa_processor
     upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
     return pipe, upscaler_2
-    
-fa_processor = FlashAttentionProcessor()
 
 pipe, upscaler_2 = load_model()