change pipeline to user StableDiffusionXLPipeling

handler.py

@@ -1,7 +1,7 @@
 from typing import Dict, List, Any
 import torch
 from torch import autocast
-from diffusers import StableDiffusionAdapterPipeline
+from diffusers import StableDiffusionXLPipeline
 import base64
 from io import BytesIO
 
@@ -13,7 +13,7 @@ if device.type != "cuda":
 
 class EndpointHandler():
   def __init__(self, path="") :
-    self.pipe = StableDiffusionAdapterPipeline.from_pretrained(path, feature_extractor=None, adapter=None, safety_checker=None, torch_dtype=torch.float16)
+    self.pipe = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
     self.pipe = self.pipe.to(device)
   
   
@@ -21,7 +21,7 @@ class EndpointHandler():
     inputs = data.pop("inputs", data)
     
     with autocast(device.type):
-      image = self.pipe(inputs, guidance_scale=7.5)["samples"][0]
+      image = self.pipe(inputs, guidance_scale=7.5).images[0]
       
     buffered = BytesIO()
     image.save(buffered, format="JPEG")