From a09f88ab0787bed82d33d9d7b68b759beb8e8e06 Mon Sep 17 00:00:00 2001
From: Fazzie <1240419984@qq.com>
Date: Wed, 16 Nov 2022 11:15:55 +0800
Subject: [PATCH] update model download in README

---
 examples/images/diffusion/README.md | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/examples/images/diffusion/README.md b/examples/images/diffusion/README.md
index be34e6d2a..08adeaeb2 100644
--- a/examples/images/diffusion/README.md
+++ b/examples/images/diffusion/README.md
@@ -46,6 +46,25 @@ pip install colossalai==0.1.10+torch1.11cu11.3 -f https://release.colossalai.org
 
 > The specified version is due to the interface incompatibility caused by the latest update of [Lightning](https://github.com/Lightning-AI/lightning), which will be fixed in the near future.
 
+## Download the model checkpoint from pretrained
+
+### stable-diffusion-v1-4
+Our default model config use the weight from [CompVis/stable-diffusion-v1-4](https://huggingface.co/CompVis/stable-diffusion-v1-4?text=A+mecha+robot+in+a+favela+in+expressionist+style)
+
+```
+git lfs install
+git clone https://huggingface.co/CompVis/stable-diffusion-v1-4
+```
+
+### stable-diffusion-v1-5 from runway
+If you want to useed the Last [stable-diffusion-v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5) wiegh from runwayml
+
+```
+git lfs install
+git clone https://huggingface.co/runwayml/stable-diffusion-v1-5
+```
+
+
 ## Dataset
 The dataSet is from [LAION-5B](https://laion.ai/blog/laion-5b/), the subset of [LAION](https://laion.ai/),
 you should the change the `data.file_path` in the `config/train_colossalai.yaml`