Bug fixes. allow for random negative prompts

2024-02-21 04:51:52 -07:00
parent 2478554c95
commit 49c41e6a5f
8 changed files with 166 additions and 6 deletions
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -287,7 +287,7 @@ class StableDiffusion:
                    load_safety_checker=False,
                    requires_safety_checker=False,
                    torch_dtype=self.torch_dtype,
-                    safety_checker=False,
+                    safety_checker=None,
                    **load_args
                ).to(self.device_torch)
            flush()