Spaces:
Running
on
Zero
Running
on
Zero
Update optimized.py
Browse files- optimized.py +2 -1
optimized.py
CHANGED
|
@@ -89,7 +89,8 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
|
|
| 89 |
# Load control image
|
| 90 |
control_image = load_image(control_image)
|
| 91 |
w, h = control_image.size
|
| 92 |
-
|
|
|
|
| 93 |
control_image = control_image.resize((int(w * scale), int(h * scale)))
|
| 94 |
print("Size to: " + str(control_image.size[0]) + ", " + str(control_image.size[1]))
|
| 95 |
image = pipe(
|
|
|
|
| 89 |
# Load control image
|
| 90 |
control_image = load_image(control_image)
|
| 91 |
w, h = control_image.size
|
| 92 |
+
w = w - w % 8
|
| 93 |
+
h = h - h % 8
|
| 94 |
control_image = control_image.resize((int(w * scale), int(h * scale)))
|
| 95 |
print("Size to: " + str(control_image.size[0]) + ", " + str(control_image.size[1]))
|
| 96 |
image = pipe(
|