a

handler.py

@@ -12,8 +12,7 @@ from diffusers.utils import load_image
 class EndpointHandler():
     def __init__(self, path=""):                
         self.pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
-        self.pipe.to("cuda")
-        self.pipe.unet = torch.compile(self.pipe.unet, mode="reduce-overhead", fullgraph=True)
+        self.pipe.to("cuda")        
         self.pipe.upcast_vae()