socialtrait
/

stable-diffusion-xl-base-1.0-infendpoint

StableDiffusionXLPipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

Suraj Narayanan Sasikumar commited on Sep 5, 2023

Commit

bba78d3

•

1 Parent(s): 2671954

using stability directly

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -28,7 +28,7 @@ class EndpointHandler:
         self.base.unet = torch.compile(self.base.unet, mode="reduce-overhead", fullgraph=True)
         self.refiner = StableDiffusionXLPipeline.from_pretrained(
-            "socialtrait/stable-diffusion-xl-refiner-1.0-infendpoint",
             text_encoder_2=self.base.text_encoder_2,
             vae=self.base.vae,
             torch_dtype=torch.float16,

         self.base.unet = torch.compile(self.base.unet, mode="reduce-overhead", fullgraph=True)
         self.refiner = StableDiffusionXLPipeline.from_pretrained(
+            "stabilityai/stable-diffusion-xl-refiner-1.0",
             text_encoder_2=self.base.text_encoder_2,
             vae=self.base.vae,
             torch_dtype=torch.float16,