Hugging Face's logo

Charles-Elena
/
ControlNet-endpoint-test

stable-diffusion
stable-diffusion-diffusers
controlnet
endpoints-template
Model card Files Community
charleselena commited on
Commit
fc91702
·
verified ·
1 Parent(s): a0c9311

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -1
handler.py CHANGED
@@ -72,7 +72,8 @@ class EndpointHandler():
72
  torch_dtype=dtype,
73
  #safety_checker=None).to(device)
74
  #processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
75
- safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
 
76
 
77
  # Define Generator with seed
78
  self.generator = torch.Generator(device="cpu").manual_seed(3)
 
72
  torch_dtype=dtype,
73
  #safety_checker=None).to(device)
74
  #processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
75
+ #safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
76
+ safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
77
 
78
  # Define Generator with seed
79
  self.generator = torch.Generator(device="cpu").manual_seed(3)