CompVis
/

stable-diffusion-safety-checker

Inference Endpoints

Model card Files Files and versions Community

Wok commited on Dec 4, 2023

Commit

24d5845

•

1 Parent(s): cb41f3a

Fix imports

Files changed (1) hide show

README.md +5 -3

README.md CHANGED Viewed

@@ -174,8 +174,10 @@ Use the code below to get started with the model.
 <summary> Click to expand </summary>
 ```python
-from transformers import AutoProcessor, SafetyChecker
-processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
-safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
  ```
 </details>

 <summary> Click to expand </summary>
 ```python
+from transformers import AutoFeatureExtractor
+from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
+processor = AutoFeatureExtractor.from_pretrained("CompVis/stable-diffusion-safety-checker")
+safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
  ```
 </details>