Spaces:
Running
on
Zero
Running
on
Zero
Update raw.py
Browse files
raw.py
CHANGED
|
@@ -68,14 +68,14 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
|
|
| 68 |
print("Size to: " + str(control_image.size[0]) + ", " + str(control_image.size[1]))
|
| 69 |
with torch.inference_mode():
|
| 70 |
image = pipe(
|
| 71 |
-
generator=generator
|
| 72 |
prompt=prompt,
|
| 73 |
control_image=control_image,
|
| 74 |
controlnet_conditioning_scale=controlnet_conditioning_scale,
|
| 75 |
num_inference_steps=steps,
|
| 76 |
guidance_scale=guidance_scale,
|
| 77 |
height=control_image.size[1],
|
| 78 |
-
width=control_image.size[0]
|
| 79 |
control_guidance_end=guidance_end
|
| 80 |
).images[0]
|
| 81 |
|
|
|
|
| 68 |
print("Size to: " + str(control_image.size[0]) + ", " + str(control_image.size[1]))
|
| 69 |
with torch.inference_mode():
|
| 70 |
image = pipe(
|
| 71 |
+
generator=generator,
|
| 72 |
prompt=prompt,
|
| 73 |
control_image=control_image,
|
| 74 |
controlnet_conditioning_scale=controlnet_conditioning_scale,
|
| 75 |
num_inference_steps=steps,
|
| 76 |
guidance_scale=guidance_scale,
|
| 77 |
height=control_image.size[1],
|
| 78 |
+
width=control_image.size[0],
|
| 79 |
control_guidance_end=guidance_end
|
| 80 |
).images[0]
|
| 81 |
|