Merge pull request #12 from bendeguzvaradi/main

Bug/Safety checker to None
2023-09-14 15:31:09 -06:00 · 2023-09-14 15:31:09 -06:00 · 3eb3535683
parent bee0b6a235 3d387103cd
commit 3eb3535683
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(