linoyts HF staff commited on
Commit
cfdcec0
1 Parent(s): b521955

pass width, height to pipeline

Browse files

width, height were sent to `infer` w/o being passed to the pipeline, changed in this PR

Files changed (1) hide show
  1. app.py +1 -0
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