Update app.py

app.py

@@ -23,6 +23,11 @@ control_modes = [
     "lq"
 ]
 
+def adjust_dimensions(width, height):
+    adjusted_width = width - (width % 8)
+    adjusted_height = height - (height % 8)
+    return adjusted_width, adjusted_height
+
 @spaces.GPU
 def generate_image(prompt, control_image_depth, control_mode_depth_index, use_depth, control_image_canny, control_mode_canny_index):
     control_images = []
@@ -39,13 +44,14 @@ def generate_image(prompt, control_image_depth, control_mode_depth_index, use_de
     conditioning_scales.append(0.4)
 
     width, height = control_image_canny.shape[:2]
+    adjusted_width, adjusted_height = adjust_dimensions(width, height)
 
     image = pipe(
         prompt,
         control_image=control_images,
         control_mode=control_modes,
-        width=width,
-        height=height,
+        width=adjusted_width,
+        height=adjusted_height,
         controlnet_conditioning_scale=conditioning_scales,
         num_inference_steps=24,
         guidance_scale=3.5,