Update handler.py

handler.py

@@ -3,7 +3,7 @@ import torch
 import base64
 from PIL import Image
 from io import BytesIO
-from diffusers import T2IAdapter, StableDiffusionXLAdapterPipeline, AutoencoderKL, DPMSolverMultistepScheduler
+from diffusers import T2IAdapter, StableDiffusionXLAdapterPipeline, StableDiffusionXLImg2ImgPipeline, AutoencoderKL, DPMSolverMultistepScheduler
 from controlnet_aux.pidi import PidiNetDetector
 
 # set device
@@ -52,10 +52,9 @@ class EndpointHandler():
         ).to("cuda")
 
         # instantiate HF refiner to improve output image
-        self.refiner = StableDiffusionXLAdapterPipeline.from_pretrained(
+        self.refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
             "stabilityai/stable-diffusion-xl-refiner-1.0",
             text_encoder_2=self.pipeline.text_encoder_2,
-            adapter=adapter,
             vae=vae,
             torch_dtype=torch.float16,
             variant="fp16",