Update app.py

app.py

@@ -27,7 +27,10 @@ import time
 import datetime
 from gradio import themes
 from image_gen_aux import UpscaleWithModel
-from diffusers.models.attention_processor import AttnProcessor2_0
+
+#from diffusers.models.attention_processor import AttnProcessor2_0
+
+from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
 
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False