Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -106,7 +106,7 @@ class EndpointHandler():
|
|
| 106 |
|
| 107 |
# hyperparameters
|
| 108 |
negative_prompt = data.pop("negative_prompt", None)
|
| 109 |
-
num_inference_steps = data.pop("num_inference_steps",
|
| 110 |
guidance_scale = data.pop("guidance_scale", 5)
|
| 111 |
negative_prompt = data.pop("negative_prompt", None)
|
| 112 |
height = data.pop("height", None)
|
|
|
|
| 106 |
|
| 107 |
# hyperparameters
|
| 108 |
negative_prompt = data.pop("negative_prompt", None)
|
| 109 |
+
num_inference_steps = data.pop("num_inference_steps", 30)
|
| 110 |
guidance_scale = data.pop("guidance_scale", 5)
|
| 111 |
negative_prompt = data.pop("negative_prompt", None)
|
| 112 |
height = data.pop("height", None)
|