From 590bf826c0027bc704bd4f5638999334a86cce1d Mon Sep 17 00:00:00 2001 From: hodanov <1031hoda@gmail.com> Date: Sun, 21 May 2023 18:49:51 +0900 Subject: [PATCH] Remove unnecessary lines. --- sd_cli.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/sd_cli.py b/sd_cli.py index c4fd228..304911b 100644 --- a/sd_cli.py +++ b/sd_cli.py @@ -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()