v1.0.2

README.md

@@ -27,7 +27,8 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk2
 docker run -it --shm-size=32G -v $PWD/MobileVLM:/home/MobileVLM -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name mobilevlm <your IMAGE ID> bash
 cd /home/MobileVLM
 pip install -r requirements.txt
-# flash_attn2、bitsandbytes可从whl.zip文件里获取安装：
+# deepspeed、flash_attn2、bitsandbytes可从whl.zip文件里获取安装：
+pip install deepspeed-0.12.3+git299681e.abi0.dtk2310.torch2.1.0a0-cp38-cp38-linux_x86_64.whl
 pip install flash_attn-2.0.4_torch2.1_dtk2310-cp38-cp38-linux_x86_64.whl
 pip install bitsandbytes-0.43.0-py3-none-any.whl
 ```
@@ -37,7 +38,8 @@ cd MobileVLM/docker
 docker build --no-cache -t mobilevlm:latest .
 docker run --shm-size=32G --name mobilevlm -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video -v $PWD/../../MobileVLM:/home/MobileVLM -it mobilevlm bash
 # 若遇到Dockerfile启动的方式安装环境需要长时间等待，可注释掉里面的pip安装，启动容器后再安装python库：pip install -r requirements.txt。
-# flash_attn2、bitsandbytes可从whl.zip文件里获取安装：
+# deepspeed、flash_attn2、bitsandbytes可从whl.zip文件里获取安装：
+pip install deepspeed-0.12.3+git299681e.abi0.dtk2310.torch2.1.0a0-cp38-cp38-linux_x86_64.whl
 pip install flash_attn-2.0.4_torch2.1_dtk2310-cp38-cp38-linux_x86_64.whl
 pip install bitsandbytes-0.43.0-py3-none-any.whl
 ```
@@ -58,6 +60,7 @@ bitsandbytes:0.43.0
 
 ```
 # flash_attn2、bitsandbytes可从whl.zip文件里获取安装：
+pip install deepspeed-0.12.3+git299681e.abi0.dtk2310.torch2.1.0a0-cp38-cp38-linux_x86_64.whl
 pip install flash_attn-2.0.4_torch2.1_dtk2310-cp38-cp38-linux_x86_64.whl
 pip install bitsandbytes-0.43.0-py3-none-any.whl
 ```