Sebastian Semeniuc commited on
Commit
466101e
·
1 Parent(s): a1f58f2

fix: fix inference endpoint

Browse files
Files changed (1) hide show
  1. 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", None)
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 if controlnet_conditioning_scale is not None else [1.0, 0.8],
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