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

Update README.md

parent bba81e48
......@@ -11,6 +11,7 @@ Codestral使用了80+的编程语言的多样化数据集进行训练,并且
<div align=center>
<img src="./assets/model_framework.png"/>
</div>
## 环境配置
-v 路径、docker_name和imageID根据实际情况修改
......@@ -66,8 +67,7 @@ export HF_ENDPOINT=https://hf-mirror.com
基于Huggingface's Transformers进行推理.<br>
模型下载后 默认需存放至weights文件夹中<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
HIP_VISIBLE_DEVICES=0 python inference.py
```
......@@ -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
└── 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