Update app.py

app.py

@@ -18,7 +18,7 @@ if is_torch2_available():
         AttnProcessor2_0 as AttnProcessor
     # from gradio_utils import SpatialAttnProcessor2_0
 else:
-    from gradio_utils  import AttnProcessor
+    from gradio_utils import AttnProcessor
 
 import diffusers
 from diffusers import StableDiffusionXLPipeline
@@ -40,7 +40,7 @@ DEFAULT_STYLE_NAME = "Japanese Anime"
 global models_dict
 use_va = True
 models_dict = {
-    "Juggernaut": "RunDiffusion/Juggernaut-XL-v8",
+#    "Juggernaut": "RunDiffusion/Juggernaut-XL-v8",
 #   "RealVision": "SG161222/RealVisXL_V4.0" ,
     "SDXL":"stabilityai/stable-diffusion-xl-base-1.0" ,
 #   "Unstable": "stablediffusionapi/sdxl-unstable-diffusers-y"
@@ -435,7 +435,7 @@ use_safetensors= False
 # # pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 # pipe1.scheduler.set_timesteps(50)
 ### 
-pipe2 = PhotoMakerStableDiffusionXLPipeline.from_pretrained(
+''''pipe2 = PhotoMakerStableDiffusionXLPipeline.from_pretrained(
     models_dict["Juggernaut"], torch_dtype=torch.float16, use_safetensors=use_safetensors)
 pipe2 = pipe2.to("cpu")
 pipe2.load_photomaker_adapter(
@@ -446,7 +446,7 @@ pipe2.load_photomaker_adapter(
 )
 pipe2 = pipe2.to("cpu")
 pipe2.enable_freeu(s1=0.6, s2=0.4, b1=1.1, b2=1.2)
-pipe2.fuse_lora()
+pipe2.fuse_lora()'''
 
 pipe4 = PhotoMakerStableDiffusionXLPipeline.from_pretrained(
     models_dict["SDXL"], torch_dtype=torch.float16, use_safetensors=True)
@@ -520,7 +520,7 @@ def process_generation(_sd_type,_model_type,_upload_images, _num_steps,style_nam
     num_steps =_num_steps
     use_safe_tensor = True
     if  style_name == "(No style)":
-        sd_model_path = models_dict["Juggernaut"]
+        sd_model_path = models_dict["SDXL"]
     if _model_type == "original":
         pipe = StableDiffusionXLPipeline.from_pretrained(sd_model_path, torch_dtype=torch.float16)
         pipe = pipe.to(device)
@@ -529,7 +529,7 @@ def process_generation(_sd_type,_model_type,_upload_images, _num_steps,style_nam
         # pipe.scheduler.set_timesteps(50)
         set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)
     elif _model_type == "Photomaker":
-        if _sd_type != "Juggernaut" and style_name != "(No style)":
+        if _sd_type != "SDXL" and style_name != "(No style)":
             pipe = pipe2.to(device)
             pipe.id_encoder.to(device)
             set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)