From c07049c1344a6b708312c5a8d9b7e3bb6c522a4a Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
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')

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-70-g09d2