fix: undefined StableDiffusionSafetyChecker
Browse files
app.py
CHANGED
@@ -21,6 +21,7 @@ from diffusers import (
|
|
21 |
DPMSolverMultistepScheduler,
|
22 |
EulerDiscreteScheduler
|
23 |
)
|
|
|
24 |
from transformers import AutoFeatureExtractor
|
25 |
import random
|
26 |
import time
|
|
|
21 |
DPMSolverMultistepScheduler,
|
22 |
EulerDiscreteScheduler
|
23 |
)
|
24 |
+
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
25 |
from transformers import AutoFeatureExtractor
|
26 |
import random
|
27 |
import time
|