Views
No views yet
pip install --upgrade diffusers # make sure to use at least diffusers >= 0.22
pip install transformers accelerate1from diffusers import DiffusionPipeline
2import torch
3
4pipe = DiffusionPipeline.from_pretrained("TobDeBer/lcm_dream7")
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").images