Sebastian Semeniuc commited on
Commit ·
466101e
1
Parent(s): a1f58f2
fix: fix inference endpoint
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -142,7 +142,7 @@ class EndpointHandler():
|
|
| 142 |
height = data.pop("height", 1024)
|
| 143 |
width = data.pop("width", 1024)
|
| 144 |
controlnet_conditioning_scale = data.pop(
|
| 145 |
-
"controlnet_conditioning_scale",
|
| 146 |
|
| 147 |
# process image
|
| 148 |
image = self.decode_base64_image(image)
|
|
@@ -161,7 +161,7 @@ class EndpointHandler():
|
|
| 161 |
num_inference_steps=num_inference_steps,
|
| 162 |
guidance_scale=guidance_scale,
|
| 163 |
num_images_per_prompt=num_of_images,
|
| 164 |
-
controlnet_conditioning_scale=controlnet_conditioning_scale
|
| 165 |
generator=self.generator
|
| 166 |
).images[0]
|
| 167 |
|
|
|
|
| 142 |
height = data.pop("height", 1024)
|
| 143 |
width = data.pop("width", 1024)
|
| 144 |
controlnet_conditioning_scale = data.pop(
|
| 145 |
+
"controlnet_conditioning_scale", 1.0)
|
| 146 |
|
| 147 |
# process image
|
| 148 |
image = self.decode_base64_image(image)
|
|
|
|
| 161 |
num_inference_steps=num_inference_steps,
|
| 162 |
guidance_scale=guidance_scale,
|
| 163 |
num_images_per_prompt=num_of_images,
|
| 164 |
+
controlnet_conditioning_scale=controlnet_conditioning_scale,
|
| 165 |
generator=self.generator
|
| 166 |
).images[0]
|
| 167 |
|