Update README.md

README.md

@@ -45,7 +45,10 @@ For more detailed instructions, use-cases and examples in JAX follow the instruc
 
 ### Original GitHub Repository
 
-1. Download the weights [v1-5-pruned-emaonly.ckpt](https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt)
+1. Download the weights 
+   - [v1-5-pruned-emaonly.ckpt](https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt) - 4.27GB, ema-only weight. uses less VRAM - suitable for inference
+   - [v1-5-pruned.ckpt](https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned.ckpt) - 7.7GB, ema+non-ema weights. uses more VRAM - suitable for training/fine-tuning
+
 2. Follow instructions [here](https://github.com/runwayml/stable-diffusion).
 
 ## Model Details