karimbenharrak commited on
Commit
5d55935
1 Parent(s): 58448df

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -39,7 +39,7 @@ class EndpointHandler():
39
  )
40
 
41
  self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
42
- "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
43
  )
44
 
45
  self.pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 
39
  )
40
 
41
  self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
42
+ "runwayml/stable-diffusion-v1-5", controlnet=self.controlnet, torch_dtype=torch.float16
43
  )
44
 
45
  self.pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)