Hugging Face's logo

enesbol
/
pix2pix_serving_

Model card Files
xet
Community
enesbol commited on
Commit
56c808e
·
1 Parent(s): 575cc5f

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -1
handler.py CHANGED
@@ -40,7 +40,9 @@ class EndpointHandler():
40
  # inference steps:
41
  num_inference_steps = info.pop("num_inference_steps", data.get("num_inference_steps", 40))
42
  # Image guidance scale
43
- image_guidance_scale = info.pop("image_guidance_scale", data.get("image_guidance_scale", 1.5))
 
 
44
  # Guidance scale
45
  guidance_scale = info.pop("guidance_scale", data.get("guidance_scale", 7.5))
46
 
 
40
  # inference steps:
41
  num_inference_steps = info.pop("num_inference_steps", data.get("num_inference_steps", 40))
42
  # Image guidance scale
43
+ image_guidance_scale=info.pop("image_guidance_scale",data)
44
+
45
+ #image_guidance_scale = info.pop("image_guidance_scale", data.get("image_guidance_scale", 1.5))
46
  # Guidance scale
47
  guidance_scale = info.pop("guidance_scale", data.get("guidance_scale", 7.5))
48