Views
No views yet


pip install diffusers transformers accelerate1from diffusers import DiffusionPipeline
2import torch
3
4pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
5
6# To save GPU memory, torch.float16 can be used, but it may compromise image quality.
7pipe.to(torch_device="cuda", torch_dtype=torch.float32)
8
9prompt = "Self-portrait oil painting, a beautiful cyborg with golden hair, 8k"
10
11# Can be set to 1~50 steps. LCM support fast inference even <= 4 steps. Recommend: 1~8 steps.
12num_inference_steps = 4
13
14images = pipe(prompt=prompt, num_inference_steps=num_inference_steps, guidance_scale=8.0, lcm_origin_steps=50, output_type="pil").images1@misc{luo2023latent,
2 title={Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inference},
3 author={Simian Luo and Yiqin Tan and Longbo Huang and Jian Li and Hang Zhao},
4 year={2023},
5 eprint={2310.04378},
6 archivePrefix={arXiv},
7 primaryClass={cs.CV}
8}