Update handler.py

handler.py

@@ -15,8 +15,8 @@ if device.type != 'cuda':
 class EndpointHandler():
     def __init__(self, path=""):
 
-        self.fast_pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to("cuda")
-        self.generator = torch.Generator(device="cuda").manual_seed(0)
+        #self.fast_pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to("cuda")
+        #self.generator = torch.Generator(device="cuda").manual_seed(0)
 
 
         self.smooth_pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
@@ -74,7 +74,7 @@ class EndpointHandler():
         else:
             image = None
             mask_image = None 
-
+        """
         if(method == "fast"):
             image = self.fast_pipe(
                 prompt=prompt,
@@ -88,6 +88,7 @@ class EndpointHandler():
             ).images[0]
     
             return image
+        """
 
         #pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to("cuda")