Commit
·
ca68472
1
Parent(s):
bd75f67
get the params right
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -69,7 +69,7 @@ class EndpointHandler():
|
|
69 |
strength = params.pop("strength", 0.8)
|
70 |
width = params.pop("width", None)
|
71 |
manual_seed = params.pop("manual_seed", -1)
|
72 |
-
logger.info(f"strength: {strength}, manual_seed: {manual_seed}, inference_steps: {num_inference_steps}, guidance_scale: {
|
73 |
out = None
|
74 |
|
75 |
generator = torch.Generator(device='cuda')
|
|
|
69 |
strength = params.pop("strength", 0.8)
|
70 |
width = params.pop("width", None)
|
71 |
manual_seed = params.pop("manual_seed", -1)
|
72 |
+
logger.info(f"strength: {strength}, manual_seed: {manual_seed}, inference_steps: {num_inference_steps}, guidance_scale: {guidance_scale}")
|
73 |
out = None
|
74 |
|
75 |
generator = torch.Generator(device='cuda')
|