Skip to content

GitLab

Commit b268f5c5 authored by dengjb's avatar dengjb
Browse files

update readme

parent d8b4fd62
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ b268f5c5
...@@ -81,7 +81,7 @@ bash ./train.sh ...@@ -81,7 +81,7 @@ bash ./train.sh
``` ```
## 推理 ## 推理
基于Huggingface's Transformers进行推理.<br> 基于Huggingface's Transformers进行推理.<br>
模型下载后[模型下载地址](https://huggingface.co/bigcode/starcoder2-7b) 默认需存放至weights文件夹中<br> 模型下载后[模型快速下载地址](http://113.200.138.88:18080/aimodels/starcoder2-7b)[模型下载地址](https://huggingface.co/bigcode/starcoder2-7b) 默认需存放至weights文件夹中<br>
也可自行更改 inference.py文件中的 model_name 参数<br> 也可自行更改 inference.py文件中的 model_name 参数<br>
```bash ```bash
HIP_VISIBLE_DEVICES=0 python inference.py HIP_VISIBLE_DEVICES=0 python inference.py
......
icon.png 0 → 100644
View file @ b268f5c5
icon.png

62.1 KB

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