karimbenharrak commited on
Commit
e6ce10c
1 Parent(s): ef050fe

Update handler.py

Browse files
Files changed (1) hide show
  1. 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