Sean Powell commited on
Commit
f0c3269
·
1 Parent(s): d34c69a

Correct text2img_guidance_scale param name.

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -25,7 +25,7 @@ class EndpointHandler:
25
  desired_output_width = int(data.get("desired_output_width", default_desired_output_width))
26
  text2img_inference_steps = int(data.get("text2img_inference_steps", default_text2img_inference_steps))
27
  img2img_inference_steps = int(data.get("img2img_inference_steps", default_img2img_inference_steps))
28
- text2img_guidance_scale = float(data.get("guidance_scale_text2img", default_text2img_guidance_scale))
29
  img2img_strength = float(data.get("img2img_strength", default_img2img_strength))
30
 
31
  print(
 
25
  desired_output_width = int(data.get("desired_output_width", default_desired_output_width))
26
  text2img_inference_steps = int(data.get("text2img_inference_steps", default_text2img_inference_steps))
27
  img2img_inference_steps = int(data.get("img2img_inference_steps", default_img2img_inference_steps))
28
+ text2img_guidance_scale = float(data.get("text2img_guidance_scale", default_text2img_guidance_scale))
29
  img2img_strength = float(data.get("img2img_strength", default_img2img_strength))
30
 
31
  print(