- handler.py +1 -1
handler.py
CHANGED
@@ -52,7 +52,7 @@ class EndpointHandler():
|
|
52 |
# run inference pipeline
|
53 |
out = self.pipe(inputs,
|
54 |
image=image,
|
55 |
-
|
56 |
num_inference_steps=num_inference_steps,
|
57 |
denoising_start_step=denoising_start,
|
58 |
denoising_end_step=denoising_end,
|
|
|
52 |
# run inference pipeline
|
53 |
out = self.pipe(inputs,
|
54 |
image=image,
|
55 |
+
strength=strength,
|
56 |
num_inference_steps=num_inference_steps,
|
57 |
denoising_start_step=denoising_start,
|
58 |
denoising_end_step=denoising_end,
|