TalHach61 commited on
Commit
2714820
1 Parent(s): 22b3efc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def generate_(prompt, negative_prompt, pose_image, num_steps, controlnet_conditi
53
  generator = torch.Generator("cuda").manual_seed(seed)
54
  images = pipe(
55
  prompt, negative_prompt=negative_prompt, image=pose_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
56
- generator=generator,
57
  ).images
58
  return images
59
 
 
53
  generator = torch.Generator("cuda").manual_seed(seed)
54
  images = pipe(
55
  prompt, negative_prompt=negative_prompt, image=pose_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
56
+ generator=generator, height=1024, width=1024,
57
  ).images
58
  return images
59