CompVis
/

stable-diffusion-safety-checker

Inference Endpoints

Model card Files Files and versions Community

Fix imports

#45

by Wok - opened Dec 4, 2023

base: refs/heads/main

←

from: refs/pr/45

Discussion Files changed

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>