Skip to content

GitLab

Commit 9b2475c5 authored by dengjb's avatar dengjb
Browse files

Update README.md

parent bba81e48
Show whitespace changes
Inline Side-by-side
Showing with 10 additions and 2 deletions
README.md
View file @ 9b2475c5
...@@ -11,6 +11,7 @@ Codestral使用了80+的编程语言的多样化数据集进行训练，并且 ...@@ -11,6 +11,7 @@ Codestral使用了80+的编程语言的多样化数据集进行训练，并且
<div align=center> <div align=center>
<img src="./assets/model_framework.png"/> <img src="./assets/model_framework.png"/>
</div> </div>
## 环境配置 ## 环境配置
-v 路径、docker_name和imageID根据实际情况修改 -v 路径、docker_name和imageID根据实际情况修改
...@@ -66,8 +67,7 @@ export HF_ENDPOINT=https://hf-mirror.com ...@@ -66,8 +67,7 @@ export HF_ENDPOINT=https://hf-mirror.com
基于Huggingface's Transformers进行推理.<br> 基于Huggingface's Transformers进行推理.<br>
模型下载后 默认需存放至weights文件夹中<br> 模型下载后 默认需存放至weights文件夹中<br>
也可自行更改 inference.py文件中的 model_name 参数<br> 也可自行更改 inference.py文件中的 model_name 参数<br>
预训练权重快速下载中心:[SCNet AIModels](http://113.200.138.88:18080/aimodels) ,项目中的预训练权下载地址[Codestral-22B-v0.1](https://huggingface.co/mistralai/Codestral-22B-v0.1) <br>
[官方下载](https://huggingface.co/mistralai/Codestral-22B-v0.1)
```python ```python
HIP_VISIBLE_DEVICES=0 python inference.py HIP_VISIBLE_DEVICES=0 python inference.py
``` ```
...@@ -88,6 +88,14 @@ prompt：Write me a function that computes fibonacci in Rust.<br> ...@@ -88,6 +88,14 @@ prompt：Write me a function that computes fibonacci in Rust.<br>
制造,能源,教育 制造,能源,教育
## 预训练权重 ## 预训练权重
预训练权重快速下载中心:[SCNet AIModels](http://113.200.138.88:18080/aimodels)
项目中的预训练权下载地址[Codestral-22B-v0.1](http://113.200.138.88:18080/aimodels/mistralai/Codestral-22B-v0.1)
[官方下载](https://huggingface.co/mistralai/Codestral-22B-v0.1)
模型目录结构如下： 模型目录结构如下：
```bash ```bash
└── Codestral-22B-v0.1 └── Codestral-22B-v0.1
......
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