From c07049c1344a6b708312c5a8d9b7e3bb6c522a4a Mon Sep 17 00:00:00 2001 From: Volpeon Date: Tue, 27 Sep 2022 14:26:45 +0200 Subject: Fix error when saving model --- pipelines/stable_diffusion/no_check.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'pipelines/stable_diffusion') 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 @@ -from diffusers import ModelMixin +from diffusers import ModelMixin, ConfigMixin import torch -class NoCheck(ModelMixin): +class NoCheck(ModelMixin, ConfigMixin): """Can be used in place of safety checker. Use responsibly and at your own risk.""" def __init__(self): -- cgit v1.2.3-54-g00ecf