summaryrefslogtreecommitdiffstats
path: root/pipelines
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2022-09-27 14:26:45 +0200
committerVolpeon <git@volpeon.ink>2022-09-27 14:26:45 +0200
commitc07049c1344a6b708312c5a8d9b7e3bb6c522a4a (patch)
tree956ee1b8abb12d40f9a57af8df53874c9c2fb973 /pipelines
parentAdded Dreambooth training script (diff)
downloadtextual-inversion-diff-c07049c1344a6b708312c5a8d9b7e3bb6c522a4a.tar.gz
textual-inversion-diff-c07049c1344a6b708312c5a8d9b7e3bb6c522a4a.tar.bz2
textual-inversion-diff-c07049c1344a6b708312c5a8d9b7e3bb6c522a4a.zip
Fix error when saving model
Diffstat (limited to 'pipelines')
-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):