Views
No views yet

1import torch
2from diffusers import StableDiffusionPipeline
3import peft
4
5repo = "IDKiro/sdxs-512-dreamshaper"
6lora_repo = "IDKiro/sdxs-512-dreamshaper-anime"
7weight_type = torch.float16 # or float32
8
9# Load model.
10pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
11pipe.unet = PeftModel.from_pretrained(pipe.unet, lora_repo)
12pipe.to("cuda")
13
14prompt = "Self-portrait oil painting, a beautiful cyborg with golden hair, 8k"
15
16# Ensure using 1 inference step and CFG set to 0.
17image = pipe(
18 prompt,
19 num_inference_steps=1,
20 guidance_scale=0
21).images[0]
22
23image.save("output.png")@article{song2024sdxs,
author = {Yuda Song, Zehao Sun, Xuanwu Yin},
title = {SDXS: Real-Time One-Step Latent Diffusion Models with Image Conditions},
journal = {arxiv},
year = {2024},
}