From b3981c99a9d6ab9b74c8cd972fad3b7c056a3406 Mon Sep 17 00:00:00 2001 From: hodanov <1031hoda@gmail.com> Date: Wed, 24 May 2023 22:19:53 +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 a6999c0..6cc570e 100644 --- a/sd_cli.py +++ b/sd_cli.py @@ -76,7 +76,6 @@ class StableDiffusion: cache_path, scheduler=scheduler, custom_pipeline="lpw_stable_diffusion", - safety_checker=None, ).to("cuda") self.pipe.enable_xformers_memory_efficient_attention() @@ -108,7 +107,6 @@ class StableDiffusion: max_embeddings_multiples=max_embeddings_multiples, ).images - # Convert to PNG bytes image_output = [] for image in images: with io.BytesIO() as buf: