语言:

• 英文 缩略图: "https://s3.amazonaws.com/moonup/production/uploads/1663756797814-62bd5f951e22ec84279820e8.png" 标签:
• 稳定扩散模型
• 稳定扩散-扩散器
• 文本生成图像 数据集:
• lambdalabs/宝可梦-BLIP标注

由Lambda Labs基于宝可梦微调的稳定扩散模型

输入文本提示即可生成专属宝可梦角色，无需任何"提示词工程"！

若想了解如何训练自己的稳定扩散变体模型，请参阅Lambda Labs提供的示例教程。

戴珍珠耳环的少女、可爱奥巴马生物、唐纳德·特朗普、鲍里斯·约翰逊、龙猫、凯蒂猫

使用指南

!pip install diffusers==0.3.0
!pip install transformers scipy ftfy

import torch
from diffusers import StableDiffusionPipeline
from torch import autocast

pipe = StableDiffusionPipeline.from_pretrained("lambdalabs/sd-pokemon-diffusers", torch_dtype=torch.float16)  
pipe = pipe.to("cuda")

prompt = "尤达大师"
scale = 10
n_samples = 4

# 有时NSFW检测器会误判宝可梦图像
# 可自行决定是否禁用安全检查（存在风险）
disable_safety = False

if disable_safety:
  def null_safety(images, **kwargs):
      return images, False
  pipe.safety_checker = null_safety

with autocast("cuda"):
  images = pipe(n_samples*[prompt], guidance_scale=scale).images

for idx, im in enumerate(images):
  im.save(f"{idx:06}.png")

模型说明

基于BLIP标注的宝可梦图像数据集，使用Lambda GPU云服务的2xA6000显卡训练约15,000步（耗时6小时，成本约10美元）。

相关链接

• Lambda扩散器
• 宝可梦标注数据集
• 扩散器格式模型权重
• 原始模型权重
• 训练代码

由Justin Pinkney（@Buntworthy）在Lambda Labs完成训练。