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 20

Commit

900d959

•

1 Parent(s): 3594afe

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -67,7 +67,7 @@ class EndpointHandler():
                                                                       controlnet=self.controlnet,
                                                                       torch_dtype=dtype,
                                                                       #safety_checker=None).to(device)
-                                                                      safety_checker = StableDiffusionSafetyChecker.from_pretrained('CompVis/stable-diffusion-safety-checker')
         # Define Generator with seed
         self.generator = torch.Generator(device="cpu").manual_seed(3)

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