Sd Pokemon Diffusers

🚀 基於寶可夢微調的Stable Diffusion模型

本項目是由 Lambda Labs 基於寶可夢數據集微調的 Stable Diffusion 模型。只需輸入文本提示，就能生成你自己的寶可夢角色，無需進行 “提示詞工程”！如果你想了解如何訓練自己的 Stable Diffusion 變體模型，可參考 Lambda Labs 的 示例。

戴珍珠耳環的女孩、可愛的奧巴馬形象、唐納德·特朗普、鮑里斯·約翰遜、龍貓、Hello Kitty

🚀 快速開始

📦 安裝指南

!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 = "Yoda"
scale = 10
n_samples = 4

# Sometimes the nsfw checker is confused by the Pokémon images, you can disable
# it at your own risk here
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")

📚 詳細文檔

🔧 技術細節

該模型在 Lambda GPU Cloud 上使用 2 塊 A6000 GPU 對 帶有 BLIP 字幕的寶可夢圖像 進行了約 15,000 步的訓練（約 6 小時，成本約 10 美元）。

🔗 相關鏈接

• Lambda Diffusers
• 帶有字幕的寶可夢數據集
• Diffusers 格式的模型權重
• 原始模型權重
• 訓練代碼

本模型由 Justin Pinkney（@Buntworthy）在 Lambda Labs 訓練完成。