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
53ae8f06
Commit
53ae8f06
authored
May 07, 2025
by
chenzk
Browse files
Update sf.md
parent
39e32cf4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
README.md
README.md
+3
-3
No files found.
README.md
View file @
53ae8f06
...
...
@@ -47,7 +47,7 @@ SSM 是用于描述这些状态表示并根据某些输入预测其下一个状
## 环境配置
### Docker(方法一)
此处提供
[
光源
](
https://www.sourcefind.cn/#/service-details
)
拉取docker镜像的地址与使用步骤,以及
[
光合
](
https://developer.
hpccube.com
/tool/
)
开发者社区深度学习库下载地址
此处提供
[
光源
](
https://www.sourcefind.cn/#/service-details
)
拉取docker镜像的地址与使用步骤,以及
[
光合
](
https://developer.
sourcefind.cn
/tool/
)
开发者社区深度学习库下载地址
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
docker run -it --shm-size=128G -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name mamba2 <your IMAGE ID> bash # <your IMAGE ID>为以上拉取的docker的镜像ID替换,本镜像为:a4dd5be0ca23
...
...
@@ -76,7 +76,7 @@ pip install e .
### Anaconda(方法三)
此处提供本地配置、编译的详细步骤,例如:
关于本项目DCU显卡所需的特殊深度学习库可从
[
光合
](
https://developer.
hpccube.com
/tool/
)
开发者社区下载安装。
关于本项目DCU显卡所需的特殊深度学习库可从
[
光合
](
https://developer.
sourcefind.cn
/tool/
)
开发者社区下载安装。
```
#DTK驱动:dtk24.04
# python:python3.10
...
...
@@ -241,7 +241,7 @@ mamba_ssm (pretrained=state-spaces/mamba2attn-2.7b), gen_kwargs: (None), limit:
运行推理时会自动连接huggingface下载模型文件,也可使用
[
镜像网站
](
https://hf-mirror.com/
)
提前下载相关模型文件到缓存目录, 使用本地修改pretrained=/path_to_model/model_name
## 源码仓库及问题反馈
-
https://developer.
hpccube.com
/codes/modelzoo/mamba2_pytorch
-
https://developer.
sourcefind.cn
/codes/modelzoo/mamba2_pytorch
## 参考资料
-
https://github.com/state-spaces/mamba
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