a

handler.py

@@ -4,17 +4,18 @@ import torch
 import base64
 from io import BytesIO
 from PIL import Image
-from diffusers import StableDiffusionXLImg2ImgPipeline
-from diffusers.utils import load_image
+# from diffusers import StableDiffusionXLImg2ImgPipeline
+# from diffusers.utils import load_image
 import numpy as np
+from diffusers import AutoPipelineForImage2Image
+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")
+    def __init__(self, path=""):                        
+        pipe = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
         self.pipe.to("cuda")        
-        self.pipe.upcast_vae()
-
+       
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """