Update app.py

app.py

@@ -145,9 +145,9 @@ 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
+    #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
     
@@ -155,8 +155,8 @@ fa_processor = FlashAttentionProcessor()
 
 pipe, upscaler_2 = load_model()
 
-#compiled_transformer = compile_transformer()
-#spaces.aoti_apply(compiled_transformer, pipe.transformer)
+compiled_transformer = compile_transformer()
+spaces.aoti_apply(compiled_transformer, pipe.transformer)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 4096