Update README.md

2 years ago · 1946db0d07
parent 494219a43a
commit 1946db0d07
1 changed files with 6 additions and 0 deletions
--- a/README.md
+++ b/README.md
@ -73,6 +73,12 @@ git clone git@github.com:THUDM/CodeGeeX.git
 cd CodeGeeX
 pip install -e .
 ```
 Or use [CodeGeeX docker](https://hub.docker.com/r/codegeex/codegeex) to quickly set up the environment (with [nvidia-docker](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker) installed):
 ```bash
 docker pull codegeex/codegeex:latest
 # To enable GPU support, clarify device ids with --device
 docker run --gpus '"device=0,1"' -it --ipc=host --name=codegeex codegeex/codegeex
 ```
 ### Model Weights