Views
No views yet

1from diffusers import DiffusionPipeline, LCMScheduler
2import torch
3
4model_id = "stabilityai/stable-diffusion-xl-base-1.0"
5lcm_lora_id = "latent-consistency/lcm-lora-sdxl"
6pipe = DiffusionPipeline.from_pretrained(model_id, variant="fp16")
7pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
8
9pipe.load_lora_weights(lcm_lora_id, adapter_name="lora")
10pipe.load_lora_weights("./cutton_doll_lora-xl.safetensors", adapter_name="doll_sdxl")
11
12pipe.set_adapters(["lora", "doll_sdxl"], adapter_weights=[1.0, 1.2])
13pipe.to(device="cuda", dtype=torch.float16)
14
15prompt = "a chibi doll, cute"
16negative_prompt = "3d render, realistic"
17
18num_images = 9
19
20for i in range(num_images):
21 img = pipe(
22 prompt=prompt,
23 negative_prompt=negative_prompt,
24 num_inference_steps=8,
25 guidance_scale=1.5,
26 ).images[0]
27
28 img.save(f"lcm_lora_{i}.png")