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): |