Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
mamba2_pytorch
Commits
e4802632
Commit
e4802632
authored
Sep 23, 2024
by
luopl
Browse files
Update README.md
parent
000d7bab
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
3 deletions
+7
-3
README.md
README.md
+7
-3
No files found.
README.md
View file @
e4802632
...
@@ -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
## 参考资料
## 参考资料
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment