Fix: Invalid import of SafetyChecker in example code
#49
by
anon-repair-bot
- opened
README.md
CHANGED
|
@@ -174,7 +174,10 @@ Use the code below to get started with the model.
|
|
| 174 |
<summary> Click to expand </summary>
|
| 175 |
|
| 176 |
```python
|
| 177 |
-
from transformers import AutoProcessor, SafetyChecker
|
|
|
|
|
|
|
|
|
|
| 178 |
processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
|
| 179 |
safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
|
| 180 |
```
|
|
|
|
| 174 |
<summary> Click to expand </summary>
|
| 175 |
|
| 176 |
```python
|
| 177 |
+
#from transformers import AutoProcessor, SafetyChecker
|
| 178 |
+
from transformers import CLIPFeatureExtractor as AutoProcessor
|
| 179 |
+
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker as SafetyChecker
|
| 180 |
+
|
| 181 |
processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
|
| 182 |
safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
|
| 183 |
```
|