fixed readme

README.md

@@ -34,7 +34,7 @@ export HF_ENDPOINT=https://hf-mirror.com
 
 ```bash
 cd docker
-docker build --no-cache -t deepseek_coder:latest .
+docker build --no-cache -t starcoder2:latest .
 docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=80G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
 
 cd /your_code_path/starcoder2_pytorch
@@ -60,8 +60,8 @@ export HF_ENDPOINT=https://hf-mirror.com
 ## 数据集
 finetune训练样例数据采用bigcode/the-stack-smol 下的子集/data/rust [下载地址](https://hf-mirror.com/datasets/bigcode/the-stack-smol)<br>
 ```angular2html
-├── data                                                                                                  │4       37.0C    59.0W      auto     300.0W     0%         0%        Normal                              
-│   ├── assembly                                                                                          │5       37.0C    62.0W      auto     300.0W     0%         0%        Normal                              
+├── data                                                                                                                            
+│   ├── assembly                                                                                                                     
 │   │   └── data.json
 │   ├── rust                                                                                          
 │   │   └── data.json  
@@ -72,7 +72,7 @@ finetune训练样例数据采用bigcode/the-stack-smol 下的子集/data/rust [
 </div>
 
 ## 训练
-### 单机四卡
+### 单机两卡
 具体参数更改请在train.sh文件中进行,以下为必要参数 <br>
 dataset_name="{数据集地址}" <br>
 model_name="{预训练模型加载地址}" <br>
@@ -81,7 +81,7 @@ bash ./train.sh
 ```
 ## 推理
 基于Huggingface's Transformers进行推理.<br>
-模型下载后 默认需存放至weights文件夹中<br>
+模型下载后[模型下载地址](https://huggingface.co/bigcode/starcoder2-7b) 默认需存放至weights文件夹中<br>
 也可自行更改 inference.py文件中的 model_name 参数<br>
 ```bash
 HIP_VISIBLE_DEVICES=0 python inference.py

docker/Dockerfile

+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310
\ No newline at end of file