MaxMilan1 commited on
Commit ·
c0a0649
1
Parent(s): d8aa11d
change
Browse files- 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 |
|