Update handler.py

handler.py

@@ -19,10 +19,10 @@ class EndpointHandler():
         #self.generator = torch.Generator(device="cuda").manual_seed(0)
 
 
-        self.smooth_pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
-            "stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
-        )
-        self.smooth_pipe.to("cuda")
+        # self.smooth_pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
+        #     "stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
+        # )
+        # self.smooth_pipe.to("cuda")
 
         
         # load StableDiffusionInpaintPipeline pipeline
@@ -59,13 +59,14 @@ class EndpointHandler():
         strength = data.pop("strength", 0.2)
         guidance_scale = data.pop("guidance_scale", 8.0)
         num_inference_steps = data.pop("num_inference_steps", 20)
-
+        """
         if(method == "smooth"):
             if encoded_image is not None:
                 image = self.decode_base64_image(encoded_image)
                 out = self.smooth_pipe(prompt, image=image).images[0]
     
                 return out
+        """
         
         # process image
         if encoded_image is not None and encoded_mask_image is not None: