From 3d387103cd3e0ba27260b137e4c64cb326daba0a Mon Sep 17 00:00:00 2001 From: bendeguzvaradi Date: Mon, 11 Sep 2023 17:05:55 +0200 Subject: [PATCH] safety checker to None --- toolkit/stable_diffusion_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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(