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
|