Spaces:
Running
on
Zero
Running
on
Zero
pass width, height to pipeline
Browse fileswidth, height were sent to `infer` w/o being passed to the pipeline, changed in this PR
app.py
CHANGED
|
@@ -32,6 +32,7 @@ def infer(control_image, seed=42, randomize_seed=False, width=1024, height=1024,
|
|
| 32 |
guidance_scale=guidance_scale,
|
| 33 |
num_inference_steps=num_inference_steps,
|
| 34 |
generator=torch.Generator("cpu").manual_seed(seed),
|
|
|
|
| 35 |
**pipe_prior_output,
|
| 36 |
).images[0]
|
| 37 |
return images, seed
|
|
|
|
| 32 |
guidance_scale=guidance_scale,
|
| 33 |
num_inference_steps=num_inference_steps,
|
| 34 |
generator=torch.Generator("cpu").manual_seed(seed),
|
| 35 |
+
width=width, height=height,
|
| 36 |
**pipe_prior_output,
|
| 37 |
).images[0]
|
| 38 |
return images, seed
|