Remove unnecessary lines.

2023-05-21 18:49:51 +09:00
parent daf84a5280
commit 590bf826c0
1 changed files with 0 additions and 2 deletions
@@ -101,8 +101,6 @@ class StableDiffusion:
            low_cpu_mem_usage=True,
            device_map="auto",
        ).to("cuda")
-        if self.pipe.safety_checker is not None:
-            self.pipe.safety_checker = lambda images, **kwargs: (images, False)
        self.pipe.enable_xformers_memory_efficient_attention()

    @method()