diff options
-rw-r--r-- | pipelines/stable_diffusion/no_check.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pipelines/stable_diffusion/no_check.py b/pipelines/stable_diffusion/no_check.py index 06c2f72..6a6b33c 100644 --- a/pipelines/stable_diffusion/no_check.py +++ b/pipelines/stable_diffusion/no_check.py | |||
@@ -1,8 +1,8 @@ | |||
1 | from diffusers import ModelMixin | 1 | from diffusers import ModelMixin, ConfigMixin |
2 | import torch | 2 | import torch |
3 | 3 | ||
4 | 4 | ||
5 | class NoCheck(ModelMixin): | 5 | class NoCheck(ModelMixin, ConfigMixin): |
6 | """Can be used in place of safety checker. Use responsibly and at your own risk.""" | 6 | """Can be used in place of safety checker. Use responsibly and at your own risk.""" |
7 | 7 | ||
8 | def __init__(self): | 8 | def __init__(self): |