TalHach61 commited on
Commit
131903f
1 Parent(s): 018a675

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ def generate_(prompt, negative_prompt, pose_image, num_steps, controlnet_conditi
89
  generator = torch.Generator("cuda").manual_seed(seed)
90
  images = pipe(
91
  prompt, negative_prompt=negative_prompt, image=pose_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
92
- generator=generator, height=1024, width=1024,
93
  ).images
94
  return images
95
 
 
89
  generator = torch.Generator("cuda").manual_seed(seed)
90
  images = pipe(
91
  prompt, negative_prompt=negative_prompt, image=pose_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
92
+ generator=generator, height=pose_image.size[1], width=pose_image.size[0],
93
  ).images
94
  return images
95