Update app.py

app.py

@@ -14,9 +14,8 @@ import numpy as np
 from PIL import Image
 import torch
 import torch._dynamo
-import diffusers
+#import diffusers
 from diffusers import AutoencoderKL, StableDiffusionXLPipeline
-
 from diffusers import EulerAncestralDiscreteScheduler
 #from diffusers import DPMSolverSDEScheduler
 
@@ -27,7 +26,7 @@ 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
 
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
@@ -186,6 +185,8 @@ def load_and_prepare_model():
     #pipe.scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1)
     #pipe.scheduler=EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear")
     # pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/Fantasy_World_XL.safetensors", adapter_name="fantasy")
+    # pipe.load_lora_weights("Keltezaa/flux_pussy_NSFW", weight_name="flux-pussy.safetensors", adapter_name="fantasy")
+    # pipe.load_lora_weights("xey/sldr_flux_nsfw_v2-studio", weight_name="sldr_flux_nsfw_v2-studio.safetensors", adapter_name="fantasy")
 
     pipe.vae.set_default_attn_processor()  # Set attention processor first
     
@@ -197,7 +198,7 @@ def load_and_prepare_model():
     
         #Some typical diffusers pipeline optimizations
     pipe.unet.to(memory_format=torch.channels_last) #Unsupported by hidet, but does not seem to make a difference if disabled.
-    pipe.enable_vae_tiling()
+    #pipe.enable_vae_tiling()
     #pipe.enable_xformers_memory_efficient_attention()