Don't pass image to text pipeline
Browse files- handler.py +0 -1
handler.py
CHANGED
|
@@ -67,7 +67,6 @@ class EndpointHandler():
|
|
| 67 |
else:
|
| 68 |
# run text pipeline
|
| 69 |
out = self.textPipe(prompt,
|
| 70 |
-
image=init_image,
|
| 71 |
num_inference_steps=num_inference_steps,
|
| 72 |
guidance_scale=guidance_scale,
|
| 73 |
num_images_per_prompt=1,
|
|
|
|
| 67 |
else:
|
| 68 |
# run text pipeline
|
| 69 |
out = self.textPipe(prompt,
|
|
|
|
| 70 |
num_inference_steps=num_inference_steps,
|
| 71 |
guidance_scale=guidance_scale,
|
| 72 |
num_images_per_prompt=1,
|