许可协议: openrail++ 标签:

• 文本生成图像
• 稳定扩散 库名称: diffusers 推理: 禁用

SDXS-512-梦幻塑造者-动漫版

SDXS是一款能够根据提示文本实时生成高分辨率图像的模型，通过分数蒸馏和特征匹配技术训练而成。 更多技术细节请参阅我们的研究论文: SDXS: 具备图像条件的实时一步潜在扩散模型。 我们作为研究成果开源了该模型。

SDXS-512-梦幻塑造者-动漫版是SDXS-512-梦幻塑造者的动漫风格LoRA适配模块。 请关注我们的代码库获取最新动态。

Diffusers使用指南

import torch
from diffusers import StableDiffusionPipeline
import peft

模型仓库 = "IDKiro/sdxs-512-dreamshaper"
Lora仓库 = "IDKiro/sdxs-512-dreamshaper-anime"
权重类型 = torch.float16     # 或 float32

# 加载模型
管道 = StableDiffusionPipeline.from_pretrained(模型仓库, torch_dtype=权重类型)
管道.unet = PeftModel.from_pretrained(管道.unet, Lora仓库)
管道.to("cuda")

提示词 = "自画像油画，金色头发的美丽赛博格，8k画质"

# 确保使用1步推理且CFG设为0
图像 = 管道(
    提示词, 
    推理步数=1, 
    引导系数=0
).images[0]

图像.save("output.png")

引用我们的工作

@article{song2024sdxs,
  作者    = {宋雨达, 孙泽浩, 尹轩梧},
  标题    = {SDXS: 具备图像条件的实时一步潜在扩散模型},
  期刊    = {arxiv},
  年份    = {2024},
}