chunzhi/stable-diffusion-cli-on-modal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove unnecessary lines.

Browse Source
This commit is contained in:
hodanov 2023-05-21 18:49:51 +09:00
parent daf84a5280
commit 590bf826c0
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sd_cli.py
View File

@ -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()