karimbenharrak
commited on
Commit
•
e6ce10c
1
Parent(s):
ef050fe
Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -98,7 +98,7 @@ class EndpointHandler():
|
|
98 |
#pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to("cuda")
|
99 |
|
100 |
# run inference pipeline
|
101 |
-
out = self.pipe(prompt=prompt, negative_prompt=negative_prompt, image=image, mask_image=mask_image, num_inference_steps=num_inference_steps)
|
102 |
|
103 |
print("1st pipeline part successful!")
|
104 |
|
|
|
98 |
#pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to("cuda")
|
99 |
|
100 |
# run inference pipeline
|
101 |
+
out = self.pipe(prompt=prompt, negative_prompt=negative_prompt, image=image, mask_image=mask_image, strength=strength, guidance_scale=guidance_scale, num_inference_steps=num_inference_steps)
|
102 |
|
103 |
print("1st pipeline part successful!")
|
104 |
|