Sd Naruto Diffusers

🚀 火影忍者擴散模型

這是由 Lambda Labs 在火影忍者相關數據上微調的Stable Diffusion模型，可根據文本生成火影忍者風格的圖像。

點擊此處體驗即時的文本生成火影忍者圖像演示！
如果你想了解如何訓練自己的Stable Diffusion變體的更多細節，請參考這個示例。

🚀 快速開始

輸入文本提示，即可生成你自己的火影忍者風格圖像！

示例展示

《權力的遊戲》轉火影忍者風格

漫威角色轉火影忍者風格

✨ 主要特性

提示工程的重要性

我們發現，提示工程確實有助於生成引人入勝且風格一致的火影忍者風格畫像。例如，使用諸如“人物姓名 忍者肖像”或“人物姓名 火影忍者風格”這樣的提示，往往能生成更接近火影忍者角色風格的圖像，包含標誌性的頭巾和其他服裝元素。

以下是一些有無提示工程的提示示例，以說明這一點：

比爾·蓋茨

未使用提示工程

使用提示工程

可愛的兔子

未使用提示工程

使用提示工程

📦 安裝指南

要在本地運行該模型，請執行以下命令：

!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-naruto-diffusers", torch_dtype=torch.float16)  
pipe = pipe.to("cuda")

prompt = "Yoda"
scale = 10
n_samples = 4

# Sometimes the nsfw checker is confused by the Naruto 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")

📚 詳細文檔

模型描述

該模型在 帶有BLIP字幕的火影忍者圖像數據集 上進行訓練，使用 Lambda GPU Cloud 上的2張A6000 GPU，訓練約30,000步（約12小時，成本約20美元）。

相關鏈接

• Lambda Diffusers
• 帶字幕的火影忍者數據集
• Diffusers格式的模型權重
• 原始模型權重
• 火影忍者Diffusers倉庫

該模型由 Eole Cervenka 在 Justin Pinkney（@Buntworthy）於 Lambda Labs 的工作基礎上訓練而成。