辛普森弥散

*稳定弥散微调，应用于与“辛普森一家”相关的图像上

如果您想了解有关如何生成您自己的带标题数据集的详细信息，请参阅此 colab

训练使用了Hugging-Face的文本到图像训练的略微修改版本 example script

关于

输入文本提示并生成卡通/辛普森样式的图像

一个漂亮的饥饿恶魔女孩，约翰·菲利普·福尔特，非常详细的绘画，马克·赖登

加尔·加朵，卡通

更多例子

examples 文件夹包含使用此模型的 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")

数据集和训练

在家用台式电脑上，使用 1xA5000 GPU，在 Runway ML's Stable-Diffusion v1.5 上进行了10000次迭代的微调，使用 BLIP captioned Simpsons images 进行了训练

由 @Norod78 训练