mrcuddle commited on
Commit
7b6d2df
·
verified ·
1 Parent(s): d483ab6

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -12,7 +12,7 @@ if device.type != 'cuda':
12
  raise ValueError("Need to run on GPU")
13
 
14
  class EndpointHandler:
15
- def __init__(self, path="mrcuddle/URPM-Inpaint-Hyper-SDXL"):
16
  """Load the SDXL Inpainting model."""
17
  self.pipeline = StableDiffusionXLInpaintPipeline.from_pretrained(
18
  path, torch_dtype=torch.float16
 
12
  raise ValueError("Need to run on GPU")
13
 
14
  class EndpointHandler:
15
+ def __init__(self, path="mrcuddle/URPM-Inpaint-SDXL"):
16
  """Load the SDXL Inpainting model."""
17
  self.pipeline = StableDiffusionXLInpaintPipeline.from_pretrained(
18
  path, torch_dtype=torch.float16