Update handler.py
Browse filesUpdated the control_image to image
- handler.py +3 -2
handler.py
CHANGED
|
@@ -102,13 +102,14 @@ class EndpointHandler():
|
|
| 102 |
|
| 103 |
# process image
|
| 104 |
image = self.decode_base64_image(image)
|
| 105 |
-
control_image = CONTROLNET_MAPPING[self.control_type]["hinter"](image)
|
| 106 |
|
| 107 |
# run inference pipeline
|
| 108 |
out = self.pipe(
|
| 109 |
prompt=prompt,
|
| 110 |
negative_prompt=negative_prompt,
|
| 111 |
-
|
|
|
|
| 112 |
num_inference_steps=num_inference_steps,
|
| 113 |
guidance_scale=guidance_scale,
|
| 114 |
num_images_per_prompt=1,
|
|
|
|
| 102 |
|
| 103 |
# process image
|
| 104 |
image = self.decode_base64_image(image)
|
| 105 |
+
#control_image = CONTROLNET_MAPPING[self.control_type]["hinter"](image)
|
| 106 |
|
| 107 |
# run inference pipeline
|
| 108 |
out = self.pipe(
|
| 109 |
prompt=prompt,
|
| 110 |
negative_prompt=negative_prompt,
|
| 111 |
+
# image=control_image,
|
| 112 |
+
image=image,
|
| 113 |
num_inference_steps=num_inference_steps,
|
| 114 |
guidance_scale=guidance_scale,
|
| 115 |
num_images_per_prompt=1,
|