Modelpublic

eolecvk/sd-naruto-diffusers

sourceHugging Faceupdated 3y agoView on Hugging Face

4likes6downloads

Model Card

Naruto diffusers (new version available here)

_Stable Diffusion fine tuned on Naruto by [Lambda Labs](https://lambdalabs.com/)._

Put in a text prompt and generate your own Naruto character, no "prompt engineering" required!

If you want to find out how to train your own Stable Diffusion variants, see this example from Lambda Labs.

"Face of President Obama smiling", "Face of President Donald Trump", "Face of President Joe Biden"

Usage

bash

!pip install diffusers==0.3.0
!pip install transformers scipy ftfy

python

import torch
from diffusers import StableDiffusionPipeline
from torch import autocast

pipe = StableDiffusionPipeline.from_pretrained("eolecvk/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")

Model description

Trained on BLIP captioned Pokémon images using 2xA6000 GPUs on Lambda GPU Cloud for around 30,000 step (about 12 hours, at a cost of about $20).

Links

Trained by Eole Cervenka after the work of Justin Pinkney (@Buntworthy) at Lambda Labs.