fix: undefined StableDiffusionSafetyChecker

#2
by yqrashawn - opened
Files changed (1) hide show
  1. app.py +1 -0
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