mrcuddle
/

URPM-Inpaint-SDXL

StableDiffusionXLInpaintPipeline

stable-diffusion-xl

stable-diffusion-xl-diffusers

Model card Files Files and versions

mrcuddle commited on Mar 21, 2025

Commit

7b6d2df

·

verified ·

1 Parent(s): d483ab6

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -12,7 +12,7 @@ if device.type != 'cuda':
     raise ValueError("Need to run on GPU")
 class EndpointHandler:
-    def __init__(self, path="mrcuddle/URPM-Inpaint-Hyper-SDXL"):
         """Load the SDXL Inpainting model."""
         self.pipeline = StableDiffusionXLInpaintPipeline.from_pretrained(
             path, torch_dtype=torch.float16

     raise ValueError("Need to run on GPU")
 class EndpointHandler:
+    def __init__(self, path="mrcuddle/URPM-Inpaint-SDXL"):
         """Load the SDXL Inpainting model."""
         self.pipeline = StableDiffusionXLInpaintPipeline.from_pretrained(
             path, torch_dtype=torch.float16