mo-di-diffusion / README.md
nitrosocke's picture
Add diffusers code example (#4)
03b34e3
|
raw
history blame
2.09 kB
metadata
license: creativeml-openrail-m

Mo Di Diffusion

This is the fine-tuned Stable Diffusion model trained on screenshots from the modern age Disney movies. Use the tokens modern disney style in your prompts for the effect.

If you enjoy this model, please check out my other models on Huggingface

Videogame Characters rendered with the model: Videogame Samples Animal Characters rendered with the model: Animal Samples Cars and Landscapes rendered with the model: Misc. Samples

🧨 Diffusers

This model can be used just like any other Stable Diffusion model. For more information, please have a look at the Stable Diffusion.

You can also export the model to ONNX, MPS and/or FLAX/JAX.

from diffusers import StableDiffusionPipeline
import torch

model_id = "nitrosocke/mo-di-diffusion"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")

prompt = "a magical princess with golden hair, modern disney style"
image = pipe(prompt).images[0]

image.save("./magical_princess.png")

Prompt and settings for Lara Croft:

modern disney lara croft Steps: 50, Sampler: Euler a, CFG scale: 7, Seed: 3940025417, Size: 512x768

Prompt and settings for Simba:

modern disney (baby simba) Negative prompt: person human Steps: 50, Sampler: Euler a, CFG scale: 7, Seed: 1355059992, Size: 512x512

This model was trained using the diffusers based dreambooth training and prior-preservation loss in 9.000 steps and using the train-text-encoder feature.