LPX55 commited on
Commit
74ddeaf
·
verified ·
1 Parent(s): f455561

Update raw.py

Browse files
Files changed (1) hide show
  1. raw.py +2 -2
raw.py CHANGED
@@ -68,14 +68,14 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
68
  print("Size to: " + str(control_image.size[0]) + ", " + str(control_image.size[1]))
69
  with torch.inference_mode():
70
  image = pipe(
71
- generator=generator
72
  prompt=prompt,
73
  control_image=control_image,
74
  controlnet_conditioning_scale=controlnet_conditioning_scale,
75
  num_inference_steps=steps,
76
  guidance_scale=guidance_scale,
77
  height=control_image.size[1],
78
- width=control_image.size[0]
79
  control_guidance_end=guidance_end
80
  ).images[0]
81
 
 
68
  print("Size to: " + str(control_image.size[0]) + ", " + str(control_image.size[1]))
69
  with torch.inference_mode():
70
  image = pipe(
71
+ generator=generator,
72
  prompt=prompt,
73
  control_image=control_image,
74
  controlnet_conditioning_scale=controlnet_conditioning_scale,
75
  num_inference_steps=steps,
76
  guidance_scale=guidance_scale,
77
  height=control_image.size[1],
78
+ width=control_image.size[0],
79
  control_guidance_end=guidance_end
80
  ).images[0]
81