justinpinkney commited on
Commit
a81267b
1 Parent(s): 7b66cc7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -0
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).