MaxMilan1 commited on
Commit
c0a0649
·
1 Parent(s): d8aa11d
Files changed (1) hide show
  1. util/text_img.py +1 -4
util/text_img.py CHANGED
@@ -47,10 +47,7 @@ def generate_image(prompt, control_image, controlnet):
47
  elif controlnet == "Depth":
48
  control_image = get_depth(control_image)
49
 
50
- image = pipe(prompt,
51
- negative_prompt=negative_prompt,
52
- image=control_image,
53
- controlnet_conditioning_scale=1.0).images[0]
54
 
55
  image2 = rembg.remove(image)
56
 
 
47
  elif controlnet == "Depth":
48
  control_image = get_depth(control_image)
49
 
50
+ image = pipe(prompt, image=control_image).images[0]
 
 
 
51
 
52
  image2 = rembg.remove(image)
53