Sd Simpsons Model

🚀 シンプソンズ拡散モデル

「シンプソンズ」関連の画像でファインチューニングされたStable Diffusionモデルで、テキストからカートゥーン調の画像を生成できます。

🚀 クイックスタート

このモデルは、「シンプソンズ」関連の画像に特化したStable Diffusionモデルです。テキストプロンプトを入力することで、カートゥーン調の画像を生成できます。独自のBlipキャプション付きデータセットを作成する詳細については、このColabを参照してください。

トレーニングは、Hugging - Faceのテキストから画像へのトレーニング用のサンプルスクリプトの一部を変更して行われました。

✨ 主な機能

テキストプロンプトを入力することで、カートゥーン調またはシンプソンズ風の画像を生成できます。

美しい空腹の悪魔少女、ジョン・フィリップ・フォルター、非常に詳細な絵、マーク・ライデン

ガル・ガドット、カートゥーン

📚 ドキュメント

より多くのサンプル画像

サンプル画像フォルダには、このモデルのckptファイルを使用してstable - diffusion - webuiで生成されたいくつかの画像が含まれています。これらの画像のEXIF情報には、生成に使用されたパラメータが含まれています。

💻 使用例

基本的な使用法

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")

🔧 技術詳細

このモデルは、Runway MLのStable - Diffusion v1.5をBLIPキャプション付きのシンプソンズ画像で10,000回のイテレーションでファインチューニングして作成されました。トレーニングは、自宅のデスクトップコンピュータ上の1xA5000 GPUを使用して行われました。

トレーニングは@Norod78によって行われました。

📄 ライセンス

このモデルは、CreativeML OpenRAIL - Mライセンスの下で提供されています。