Commit e4802632 authored by luopl's avatar luopl
Browse files

Update README.md

parent 000d7bab
...@@ -127,9 +127,6 @@ openbookqa/ ...@@ -127,9 +127,6 @@ openbookqa/
## 推理 ## 推理
运行推理时会自动连接huggingface下载模型文件,也可使用[镜像网站](https://hf-mirror.com/)提前下载相关模型文件到缓存目录, 使用本地修改pretrained=/path_to_model/model_name
模型权重SCNet下载链接[models](http://113.200.138.88:18080/aimodels/state-spaces)
### 单机单卡 ### 单机单卡
...@@ -138,6 +135,7 @@ Evaluate: ...@@ -138,6 +135,7 @@ Evaluate:
To run evaluations on Mamba-1 models To run evaluations on Mamba-1 models
``` ```
export HF_ENDPOINT=https://hf-mirror.com
lm_eval --model mamba_ssm --model_args pretrained=state-spaces/mamba-130m --tasks lambada_openai,arc_easy,arc_challenge,winogrande,openbookqa --device cuda --batch_size 256 lm_eval --model mamba_ssm --model_args pretrained=state-spaces/mamba-130m --tasks lambada_openai,arc_easy,arc_challenge,winogrande,openbookqa --device cuda --batch_size 256
``` ```
...@@ -236,6 +234,12 @@ mamba_ssm (pretrained=state-spaces/mamba2attn-2.7b), gen_kwargs: (None), limit: ...@@ -236,6 +234,12 @@ mamba_ssm (pretrained=state-spaces/mamba2attn-2.7b), gen_kwargs: (None), limit:
`智能问答` `智能问答`
### 热点应用行业 ### 热点应用行业
`科研,制造,医疗,家居,教育` `科研,制造,医疗,家居,教育`
## 预训练权重
运行推理时会自动连接huggingface下载模型文件,也可使用[镜像网站](https://hf-mirror.com/)提前下载相关模型文件到缓存目录, 使用本地修改pretrained=/path_to_model/model_name
模型权重SCNet下载链接[models](http://113.200.138.88:18080/aimodels/state-spaces)
## 源码仓库及问题反馈 ## 源码仓库及问题反馈
- https://developer.hpccube.com/codes/modelzoo/mamba2_pytorch - https://developer.hpccube.com/codes/modelzoo/mamba2_pytorch
## 参考资料 ## 参考资料
......
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