width/height

app.py

@@ -51,6 +51,8 @@ def generate_image(
     prompt,
     ckpt,
     num_inference_steps,
+    width,
+    height,
     progress=gr.Progress(track_tqdm=True),
     mode="sdxl",
 ):
@@ -77,7 +79,7 @@ def generate_image(
             )
 
     results = pipe(
-        prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale
+        prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale, width=width, height=height
     )
 
 #    if SAFETY_CHECKER:
@@ -104,7 +106,7 @@ css = """
 with gr.Blocks(css=css) as demo:
     gr.Markdown(
         """
-# Phased Consistency Model  
+# Phased Consistency Model
 
 Phased Consistency Model (PCM) is an image generation technique that addresses the limitations of the Latent Consistency Model (LCM) in high-resolution and text-conditioned image generation.
 PCM outperforms LCM across various generation settings and achieves state-of-the-art results in both image and video generation.
@@ -118,16 +120,32 @@ PCM outperforms LCM across various generation settings and achieves state-of-the
             ckpt = gr.Dropdown(
                 label="Select inference steps",
                 choices=list(checkpoints.keys()),
-                value="4-Step",
+                value="2-Step",
             )
             steps = gr.Slider(
                 label="Number of Inference Steps",
                 minimum=1,
                 maximum=20,
                 step=1,
-                value=4,
+                value=2,
                 interactive=False,
             )
+            width = gr.Slider(
+                label="Width",
+                minimum=512,
+                maximum=1024,
+                step=256,
+                value=512,
+                interactive=True
+            )
+            height = gr.Slider(
+                label="Height",
+                minimum=512,
+                maximum=1024,
+                step=256,
+                value=512,
+                interactive=True
+            )
             ckpt.change(
                 fn=update_steps,
                 inputs=[ckpt],
@@ -169,22 +187,22 @@ PCM outperforms LCM across various generation settings and achieves state-of-the
                 4,
             ],
         ],
-        inputs=[prompt, ckpt, steps],
+        inputs=[prompt, ckpt, steps, width, height],
         outputs=[img],
         fn=generate_image,
-        cache_examples="lazy",
+        #cache_examples="lazy",
     )
 
     gr.on(
         fn=generate_image,
         triggers=[ckpt.change, prompt.submit, submit_sdxl.click],
-        inputs=[prompt, ckpt, steps],
+        inputs=[prompt, ckpt, steps, width, height],
         outputs=[img],
     )
     gr.on(
         fn=lambda *args: generate_image(*args, mode="sd15"),
         triggers=[submit_sd15.click],
-        inputs=[prompt, ckpt, steps],
+        inputs=[prompt, ckpt, steps, width, height],
         outputs=[img],
     )