Update env

README.md

@@ -18,6 +18,14 @@ The models are prepared by [Visual Informatics Group @ University of Texas at Au
 
 License: [MIT License](https://opensource.org/license/mit/)
 
+Setup environment
+```shell
+pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+pip install transformers==4.31.0
+pip install huggingface_hub accelerate
+```
+```
+
 How to use
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer