Views
No views yet
1pip install --upgrade pip
2pip install --upgrade diffusers transformers accelerate peft
3pip install matplotlib1import torch
2from diffusers import DiffusionPipeline, LCMScheduler
3import matplotlib.pyplot as plt
4
5pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0",
6 variant="fp16",
7 torch_dtype=torch.float16
8 ).to("cuda")
9# set scheduler
10pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
11
12# load LoRAs
13pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl", adapter_name="lcm")
14pipe.load_lora_weights("ming-yang/sdxl_chinese_ink_lora", adapter_name="Chinese Ink")
15
16# Combine LoRAs
17pipe.set_adapters(["lcm", "Chinese Ink"], adapter_weights=[1.0, 0.8])
18
19prompts = ["Chinese Ink, mona lisa picture, 8k", "mona lisa, 8k"]
20generator = torch.manual_seed(1)
21images = [pipe(prompt, num_inference_steps=8, guidance_scale=1, generator=generator).images[0] for prompt in prompts]
22
23fig, axs = plt.subplots(1, 2, figsize=(40, 20))
24
25axs[0].imshow(images[0])
26axs[0].axis('off') # 不显示坐标轴
27
28axs[1].imshow(images[1])
29axs[1].axis('off')
30plt.show()Chinese Ink to trigger the image generation.