Update app.py

app.py

@@ -13,7 +13,7 @@ theme = gr.themes.Base(
 )
 
 pipe = StableDiffusionXLPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
+    "corcelio/mobius",
     custom_pipeline="multimodalart/sdxl_perturbed_attention_guidance",
     torch_dtype=torch.float16
 )
@@ -33,10 +33,10 @@ def run(prompt, negative_prompt=None, guidance_scale=7.0, pag_scale=3.0, pag_lay
         guidance_scale = 0.0
     print(f"Seed before sending to generator for prompt: `{prompt}`", seed)
     generator = torch.Generator(device="cuda").manual_seed(seed)
-    image_pag = pipe(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, pag_scale=pag_scale, pag_applied_layers=pag_layers, generator=generator, num_inference_steps=25).images[0]    
+    image_pag = pipe(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, pag_scale=pag_scale, pag_applied_layers=pag_layers, generator=generator, num_inference_steps=25, clip_skip=3).images[0]    
     
     generator = torch.Generator(device="cuda").manual_seed(seed)
-    image_normal = pipe(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, generator=generator, num_inference_steps=25).images[0]
+    image_normal = pipe(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, generator=generator, num_inference_steps=25, clip_skip=3).images[0]
     print(f"Seed at the end of generation for prompt: `{prompt}`", seed)
     return (image_pag, image_normal), seed