Sd Simpsons Model

🚀 辛普森擴散模型

基於“辛普森一家”相關圖像對Stable Diffusion進行微調的模型，可根據文本提示生成卡通風格圖像

🚀 快速開始

如果你想了解如何生成自己的BLIP字幕數據集的更多細節，請查看這個Colab筆記本。

訓練使用的是經過輕微修改的Hugging Face文本到圖像訓練示例腳本。

✨ 主要特性

輸入文本提示，即可生成卡通風格/辛普森風格的圖像。

💻 使用示例

基礎用法

from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler
import torch

# this will substitute the default PNDM scheduler for K-LMS  
lms = LMSDiscreteScheduler(
    beta_start=0.00085, 
    beta_end=0.012, 
    beta_schedule="scaled_linear"
)

guidance_scale=9
seed=7777
steps=100

model_id = "Norod78/sd-simpsons-model"
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=lms, torch_dtype=torch.float16)
pipe.to("cuda")

prompt = "a photo of an astronaut riding a horse on mars"
image = pipe(prompt).images[0]  
    
image.save("astronaut_rides_horse.png")

📚 詳細文檔

示例展示

示例一：一個美麗飢餓的惡魔女孩，約翰·菲利普·法爾特風格，非常精細的畫作，馬克·賴登風格

示例二：蓋爾·加朵，卡通風格

更多示例

示例文件夾中包含了一些使用此模型的ckpt文件，通過stable-diffusion-webui生成的圖像。這意味著這些圖像的EXIF信息中包含了生成它們所使用的參數。

🔧 技術細節

該模型在我的家用桌面計算機上，使用1xA5000 GPU，基於Runway ML的Stable-Diffusion v1.5，在帶有BLIP字幕的辛普森圖像數據集上進行了10,000次迭代的微調。

訓練者：@Norod78

📄 許可證

本模型採用CreativeML OpenRAIL-M許可證。