justinpinkney
commited on
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).
|