Commit 33a7299d authored by luopl's avatar luopl
Browse files

Update README.md

parent 12c99d4b
......@@ -17,9 +17,8 @@ Step 3.7 Flash 是一个拥有 1980 亿参数的稀疏混合专家(MoE)视
|:------------:|:-----------------------------------------------:|
| DTK | 26.04 |
| python | 3.10.12 |
| vllm | 0.18.1+das.dtk2604.torch2100.2605072017.gfa7180 |
| vllm-hcu | 0.18.1+das.dtk2604.torch2100.2605072017.gfa7180 |
| transformers | 5.2.0 |
| vLLM | 0.18.1+das.dtk2604.torch2100.2605072017.gfa7180 |
| Transformers | 5.2.0 |
当前仅支持定制镜像: harbor.sourcefind.cn:5443/dcu/admin/base/custom:vllm0.18.1-ubuntu22.04-dtk26.04-py3.10-20260529-step3p7
......@@ -45,9 +44,9 @@ docker run -it \
## 预训练权重
**请根据`支持的DCU型号`选择对应模型下载,FP8模型仅在BW1100/BW1101上支持,其他型号请勿使用!**
| 模型名称 | 权重大小 | 数据类型 | DCU型号 | 最低卡数需求 | 下载地址 |
| 模型名称 | 权重大小 | 数据类型 | 支持的DCU型号 | 最低卡数需求 | 下载地址 |
|:-----:|:----------:|:----------:|:----------:|:---------------------:|:----------:|
| Step-37-Flash | 201B | BF16 | BW1100 | 8 | [ModelScope](https://www.modelscope.cn/collections/stepfun-ai/Step-37-Flash) |
| Step-3.7-Flash | 201B | BF16 | BW1100 | 8 | [ModelScope](https://www.modelscope.cn/collections/stepfun-ai/Step-37-Flash) |
| Step-3.7-Flash-FP8 | 201B | FP8 | BW1100 | 4 | [ModelScope](https://www.modelscope.cn/models/stepfun-ai/Step-3.7-Flash-FP8) |
......
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