diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py
index d74e4c7..ddeff4f 100644
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -212,7 +212,7 @@ class StableDiffusion:
                     device=self.device_torch,
                     load_safety_checker=False,
                     requires_safety_checker=False,
-                    safety_checker=False
+                    safety_checker=None
                 ).to(self.device_torch)
             else:
                 pipe = pipln.from_single_file(