Commit ·
a81267b
1
Parent(s): 7b66cc7
Update README.md
Browse files
README.md
CHANGED
|
@@ -21,6 +21,22 @@ If you want to find out how to train your own Stable Diffusion variants, see thi
|
|
| 21 |
|
| 22 |
> Girl with a pearl earring, Cute Obama creature, Donald Trump, Boris Johnson, Totoro, Hello Kitty
|
| 23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
## Model description
|
| 25 |
|
| 26 |
Trained on [BLIP captioned Pokémon images](https://huggingface.co/datasets/lambdalabs/pokemon-blip-captions) using 2xA6000 GPUs on [Lambda GPU Cloud](https://lambdalabs.com/service/gpu-cloud) for around 15,000 step (about 6 hours, at a cost of about $10).
|
|
|
|
| 21 |
|
| 22 |
> Girl with a pearl earring, Cute Obama creature, Donald Trump, Boris Johnson, Totoro, Hello Kitty
|
| 23 |
|
| 24 |
+
## Usage
|
| 25 |
+
|
| 26 |
+
Make sure you have setup the Stable Diffusion repo and downloaded `ema-only-epoch=000142.ckpt`
|
| 27 |
+
|
| 28 |
+
```bash
|
| 29 |
+
python scripts/txt2img.py \
|
| 30 |
+
--prompt 'robotic cat with wings' \
|
| 31 |
+
--outdir 'outputs/generated_pokemon' \
|
| 32 |
+
--H 512 --W 512 \
|
| 33 |
+
--n_samples 4 \
|
| 34 |
+
--config 'configs/stable-diffusion/pokemon.yaml' \
|
| 35 |
+
--ckpt ema-only-epoch=000142.ckpt
|
| 36 |
+
```
|
| 37 |
+
|
| 38 |
+
You can also use the normal stable diffusion inference config.
|
| 39 |
+
|
| 40 |
## Model description
|
| 41 |
|
| 42 |
Trained on [BLIP captioned Pokémon images](https://huggingface.co/datasets/lambdalabs/pokemon-blip-captions) using 2xA6000 GPUs on [Lambda GPU Cloud](https://lambdalabs.com/service/gpu-cloud) for around 15,000 step (about 6 hours, at a cost of about $10).
|