saassa commited on
Commit
81fc35d
1 Parent(s): 1848901

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -64,7 +64,7 @@ class EndpointHandler():
64
 
65
  # Load StableDiffusionControlNetPipeline
66
  self.stable_diffusion_id = "stablediffusionapi/disney-pixar-cartoon"
67
- self.pipe = AutoPipelineForText2Image.from_pretrained(self.stable_diffusion_id,
68
  controlnet=self.controlnet,
69
  torch_dtype=dtype,
70
  safety_checker=None).to(device)
 
64
 
65
  # Load StableDiffusionControlNetPipeline
66
  self.stable_diffusion_id = "stablediffusionapi/disney-pixar-cartoon"
67
+ self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
68
  controlnet=self.controlnet,
69
  torch_dtype=dtype,
70
  safety_checker=None).to(device)