Update app.py

app.py

@@ -50,6 +50,10 @@ def infer(
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
+    # Ensure width and height are divisible by 8
+    width = max(256, (width // 8) * 8)
+    height = max(256, (height // 8) * 8)
+
     generator = torch.Generator().manual_seed(seed)
 
     image = pipe(
@@ -129,7 +133,7 @@ with gr.Blocks(css=css) as demo:
                 label="Width",
                 minimum=256,
                 maximum=MAX_IMAGE_SIZE,
-                step=32,
+                step=8,
                 value=512,
             )
 
@@ -137,7 +141,7 @@ with gr.Blocks(css=css) as demo:
                 label="Height",
                 minimum=256,
                 maximum=MAX_IMAGE_SIZE,
-                step=32,
+                step=8,
                 value=512,
             )
 
@@ -163,4 +167,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()