Skip to content

GitLab

"...text-generation-inference.git" did not exist on "408dbc485c112e023a97fc959d4cb3721cf89fed"
Commit 260485b2 authored by dengjb's avatar dengjb
Browse files

update codes

parent 96495117
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ 260485b2
...@@ -84,7 +84,7 @@ bash ./train.sh ...@@ -84,7 +84,7 @@ bash ./train.sh
## 推理 ## 推理
基于Huggingface's Transformers进行推理.<br> 基于Huggingface's Transformers进行推理.<br>
模型下载后[模型下载地址](https://hf-mirror.com/meta-llama/CodeLlama-7b-Instruct-hf) 默认需存放至weights文件夹中<br> 模型下载后[模型下载地址](https://hf-mirror.com/meta-llama/CodeLlama-7b-Instruct-hf) [快速模型下载地址](http://113.200.138.88:18080/aimodels/CodeLlama-7b-Instruct-hf) 默认需存放至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
......
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