Update handler.py
#2
by
charleselena
- opened
- 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,
|