Commit d1bc5ff3 authored by dengjb's avatar dengjb
Browse files

Update README.md

parent b268f5c5
......@@ -81,7 +81,8 @@ bash ./train.sh
```
## 推理
基于Huggingface's Transformers进行推理.<br>
模型下载后[模型快速下载地址](http://113.200.138.88:18080/aimodels/starcoder2-7b)[模型下载地址](https://huggingface.co/bigcode/starcoder2-7b) 默认需存放至weights文件夹中<br>
预训练权重快速下载中心:[SCNet AIModels](http://113.200.138.88:18080/aimodels) ,项目中的预训练权下载地址[starcoder2-7b](http://113.200.138.88:18080/aimodels/starcoder2-7b)<br>
[starcoder2-7b](https://huggingface.co/bigcode/starcoder2-7b) 默认需存放至weights文件夹中<br>
也可自行更改 inference.py文件中的 model_name 参数<br>
```bash
HIP_VISIBLE_DEVICES=0 python inference.py
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment