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-24 22:19:53 +09:00
parent e313420b28
commit b3981c99a9
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

@ -76,7 +76,6 @@ class StableDiffusion:
cache_path, cache_path,
scheduler=scheduler, scheduler=scheduler,
custom_pipeline="lpw_stable_diffusion", custom_pipeline="lpw_stable_diffusion",
safety_checker=None,
).to("cuda") ).to("cuda")
self.pipe.enable_xformers_memory_efficient_attention() self.pipe.enable_xformers_memory_efficient_attention()
@ -108,7 +107,6 @@ class StableDiffusion:
max_embeddings_multiples=max_embeddings_multiples, max_embeddings_multiples=max_embeddings_multiples,
).images ).images
# Convert to PNG bytes
image_output = [] image_output = []
for image in images: for image in images:
with io.BytesIO() as buf: with io.BytesIO() as buf: