scheduler

app.py

@@ -7,7 +7,13 @@ import math
 import torch
 import numpy as np
 
-from diffusers import DiffusionPipeline
+from diffusers.pipelines.stable_diffusion_xl.pipeline_stable_diffusion_xl import (
+    StableDiffusionXLPipeline,
+)
+from diffusers.schedulers.scheduling_euler_ancestral_discrete import (
+    EulerAncestralDiscreteScheduler,
+)
+from diffusers.models.attention_processor import AttnProcessor2_0
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 import gradio as gr
@@ -43,11 +49,14 @@ dart = dart.requires_grad_(False)
 dart = torch.compile(dart)
 tokenizer = AutoTokenizer.from_pretrained(DART_V3_REPO_ID)
 
-pipe = DiffusionPipeline.from_pretrained(
+pipe = StableDiffusionXLPipeline.from_pretrained(
     IMAGE_MODEL_REPO_ID, 
     torch_dtype=torch_dtype,
+    add_watermarker=False,
     custom_pipeline="lpw_stable_diffusion_xl"
 )
+pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+pipe.unet.set_attn_processor(AttnProcessor2_0())
 pipe = pipe.to(device)
 
 
@@ -189,16 +198,14 @@ with gr.Blocks(css=css) as demo:
         with gr.Accordion("Advanced Settings", open=False):
             prompt_suffix = gr.Text(
                 label="Prompt suffix",
-                max_lines=1,
-                visible=False,
-                value="masterpiece, best quality",
+                visible=True,
+                value="masterpiece, best quality, very aesthetic",
             )
             negative_prompt = gr.Text(
                 label="Negative prompt",
-                max_lines=1,
                 placeholder="Enter a negative prompt",
-                visible=False,
-                value="worst quality, bad quality, low quality, lowres, displeasing, very displeasing, bad anatomy, bad hands, scan artifacts, signature, username, jpeg artifacts, guro, extra digits, fewer digits",
+                visible=True,
+                value="(worst quality, bad quality, low quality:1.2), lowres, displeasing, very displeasing, bad anatomy, bad hands, extra digits, fewer digits, scan artifacts, signature, username, jpeg artifacts, retro, 2010s",
             )
 
             seed = gr.Slider(
@@ -214,18 +221,18 @@ with gr.Blocks(css=css) as demo:
             with gr.Row():
                 width = gr.Slider(
                     label="Width",
-                    minimum=256,
+                    minimum=512,
                     maximum=MAX_IMAGE_SIZE,
-                    step=32,
-                    value=1024,  # Replace with defaults that work for your model
+                    step=64,
+                    value=832,  # Replace with defaults that work for your model
                 )
 
                 height = gr.Slider(
                     label="Height",
-                    minimum=256,
+                    minimum=512,
                     maximum=MAX_IMAGE_SIZE,
-                    step=32,
-                    value=1024,  # Replace with defaults that work for your model
+                    step=64,
+                    value=1152,  # Replace with defaults that work for your model
                 )
 
             with gr.Row():
@@ -234,15 +241,15 @@ with gr.Blocks(css=css) as demo:
                     minimum=1.0,
                     maximum=10.0,
                     step=0.5,
-                    value=6.5,
+                    value=7,
                 )
 
                 num_inference_steps = gr.Slider(
                     label="Number of inference steps",
-                    minimum=1,
+                    minimum=20,
                     maximum=50,
                     step=1,
-                    value=20,
+                    value=25,
                 )
 
     gr.on(