diff options
| author | Volpeon <git@volpeon.ink> | 2022-09-27 14:26:45 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-09-27 14:26:45 +0200 |
| commit | c07049c1344a6b708312c5a8d9b7e3bb6c522a4a (patch) | |
| tree | 956ee1b8abb12d40f9a57af8df53874c9c2fb973 /pipelines | |
| parent | Added Dreambooth training script (diff) | |
| download | textual-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.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): |