metadata
language:
- en
thumbnail: >-
https://s3.amazonaws.com/moonup/production/uploads/1663756797814-62bd5f951e22ec84279820e8.png
tags:
- stable-diffusion
- stable-diffusion-diffusers
- text-to-image
datasets:
- lambdalabs/pokemon-blip-captions
inference: false
Stable Diffusion fine tuned on Pokémon by Lambda Labs.
Put in a text prompt and generate your own Pokémon 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.
Girl with a pearl earring, Cute Obama creature, Donald Trump, Boris Johnson, Totoro, Hello Kitty
Usage
Make sure you have setup the Stable Diffusion repo and downloaded ema-only-epoch=000142.ckpt
python scripts/txt2img.py \
--prompt 'robotic cat with wings' \
--outdir 'outputs/generated_pokemon' \
--H 512 --W 512 \
--n_samples 4 \
--config 'configs/stable-diffusion/pokemon.yaml' \
--ckpt ema-only-epoch=000142.ckpt
You can also use the normal stable diffusion inference config.
Model description
Trained on BLIP captioned Pokémon images using 2xA6000 GPUs on Lambda GPU Cloud for around 15,000 step (about 6 hours, at a cost of about $10).
Links
- Lambda Diffusers
- Captioned Pokémon dataset
- Model weights in Diffusers format
- Original model weights
- Training code
Trained by Justin Pinkney (@Buntworthy) at Lambda Labs.