Update README.md

README.md

@@ -15,38 +15,57 @@ license: llama2
 - NVIDIA GPU + [CUDA](https://developer.nvidia.com/cuda-downloads)
 
 ### Installation
-1. Clone repo
+Clone the repo and install dependent packages
 
-    ```bash
-    git clone https://github.com/AILab-CVC/SEED.git
-    cd SEED
-    ```
+  ```bash
+  git clone https://github.com/AILab-CVC/SEED.git
+  cd SEED
+  pip install -r requirements.txt
+  ```
 
-2. Install dependent packages
-
-    ```bash
-    pip install -r requirements.txt
-    ``` 
     
 ### Model Weights
-We provide the pretrained SEED Tokenizer and De-Tokenizer, instruction tuned SEED-LLaMA-8B and SEED-LLaMA-14B. 
+We release the pretrained SEED Tokenizer and De-Tokenizer, instruction tuned SEED-LLaMA-8B and SEED-LLaMA-14B in [SEED Hugging Face](https://huggingface.co/AILab-CVC/SEED). 
 Please download the checkpoints and save under the folder `./pretrained`.
 
+```bash
+cd pretrained   # SEED/pretrained
+git lfs install
+git clone https://huggingface.co/AILab-CVC/SEED
+mv SEED/* ./
+```
+
 To reconstruct the image from the SEED visual codes using unCLIP SD-UNet, please download the pretrained [unCLIP SD](https://huggingface.co/stabilityai/stable-diffusion-2-1-unclip).
 Rename the checkpoint directory to **"diffusion_model"** and create a soft link to the "pretrained/seed_tokenizer" directory.
 
+```bash
+# SEED/pretrained
+git lfs install
+git clone https://huggingface.co/stabilityai/stable-diffusion-2-1-unclip
+mv stable-diffusion-2-1-unclip seed_tokenizer/diffusion_model
+```
+
 
 ### Inference for visual tokenization and de-tokenization
 To discretize an image to 1D visual codes with causal dependency, and reconstruct the image from the visual codes using the off-the-shelf unCLIP SD-UNet:
+
 ```bash
+cd ..   # SEED/ 
 python scripts/seed_tokenizer_inference.py
 ```
 
-### Launching Demo of SEED-LLaMA Locally 
+### Launching Gradio Demo of SEED-LLaMA-14B Locally 
+Building the local demo of SEED-LLaMA-14B currently requires 2*32GB devices.
+
 ```bash
-sh start_backend.sh
-sh start_frontend.sh
+# SEED/
+# in first terminal
+sh scripts/start_backend.sh
+# in second terminal
+sh scripts/start_frontend.sh
 ``` 
+Then the demo can be accessed through http://127.0.0.1:80
+
 
 ## Citation
 If you find the work helpful, please consider citing: