summaryrefslogtreecommitdiffstats
path: root/pipelines/stable_diffusion
diff options
context:
space:
mode:
Diffstat (limited to 'pipelines/stable_diffusion')
-rw-r--r--pipelines/stable_diffusion/no_check.py4
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 @@
1from diffusers import ModelMixin 1from diffusers import ModelMixin, ConfigMixin
2import torch 2import torch
3 3
4 4
5class NoCheck(ModelMixin): 5class 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):