Charles-Elena
/

ControlNet-endpoint-test

stable-diffusion

stable-diffusion-diffusers

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

charleselena commited on Jan 21, 2024

Commit

d0be649

·

verified ·

1 Parent(s): 081955a

replace dtype

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -89,7 +89,7 @@ class EndpointHandler():
         self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
                                                                       controlnet=self.controlnet,
                                                                       torch_dtype=dtype,
-                                                                      safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker", torch_dtype=torch.float16)).to("cuda")
         # Define Generator with seed
         self.generator = torch.Generator(device=device.type).manual_seed(3)

         self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
                                                                       controlnet=self.controlnet,
                                                                       torch_dtype=dtype,
+                                                                      safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker", torch_dtype=dtype)).to("cuda")
         # Define Generator with seed
         self.generator = torch.Generator(device=device.type).manual_seed(3)