safety checker to None

2026-02-08 14:39:58 +00:00 · 2023-09-11 17:05:55 +02:00
parent bee0b6a235
commit 3d387103cd
1 changed files with 1 additions and 1 deletions
--- 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(