语言:

• 英文 标签:
• 稳定扩散模型
• 文本生成图像 许可证: bigscience-bloom-rail-1.0 推理支持: 是 缩略图: "https://cdn.discordapp.com/attachments/1020199895694589953/1020200601780494386/000001.553325548.png"

小马扩散模型 - >无蹄版

小马扩散模型V4现已发布!

小马扩散模型是基于高质量小马安全向图像进行微调的潜在文本生成图像扩散模型。

特别鸣谢Waifu-Diffusion提供的微调技术支持，以及Novel AI提供的算力支持。

原始PyTorch模型下载链接

针对小马面部优化的Real-ESRGAN模型

模型描述

本模型的微调基础是Waifu-Diffusion的早期微调检查点，其本身基于Stable Diffusion V1-4——一个在LAION2B-en数据集上训练的潜在图像扩散模型。

该特定检查点以5.0e-6的学习率，在约8万张评分超过500且属于"safe"或"suggestive"分类的小马图文配对数据（使用derpibooru标签）上进行了4个epoch的微调。

许可证

本模型采用CreativeML OpenRAIL-M许可证向公众开放，该许可证进一步明确了使用权限：

1. 禁止使用模型故意生成或传播非法/有害内容
2. 模型作者对生成内容不主张权利，使用者需自行承担内容责任并遵守许可条款
3. 允许商业用途和模型再分发，但必须包含原始许可条款并向所有用户提供完整的CreativeML OpenRAIL-M许可证副本 完整许可证文本

下游应用

适用于娱乐用途和作为生成艺术创作辅助工具。

示例代码

import torch
from torch import autocast
from diffusers import StableDiffusionPipeline, DDIMScheduler
model_id = "AstraliteHeart/pony-diffusion"
device = "cuda"
pipe = StableDiffusionPipeline.from_pretrained(
    model_id,
    torch_dtype=torch.float16,
    revision="fp16",
    scheduler=DDIMScheduler(
        beta_start=0.00085,
        beta_end=0.012,
        beta_schedule="scaled_linear",
        clip_sample=False,
        set_alpha_to_one=False,
    ),
)
pipe = pipe.to(device)
prompt = "粉红派拟人肖像 婚纱头纱 精致细节 数字绘画 artstation概念艺术 平滑锐焦 插画 Unreal Engine 5 8K"
with autocast("cuda"):
    image = pipe(prompt, guidance_scale=7.5)["sample"][0]  
    
image.save("可爱小马.png")

开发团队与致谢

本项目的实现离不开CompVis研究团队的开创性工作。

特别感谢：

• Waifu-Diffusion提供微调技术支持和基础检查点
• Novel AI提供算力支持

联系我们请加入Discord服务器