Commit 53ae8f06 authored by chenzk's avatar chenzk
Browse files

Update sf.md

parent 39e32cf4
...@@ -47,7 +47,7 @@ SSM 是用于描述这些状态表示并根据某些输入预测其下一个状 ...@@ -47,7 +47,7 @@ SSM 是用于描述这些状态表示并根据某些输入预测其下一个状
## 环境配置 ## 环境配置
### Docker(方法一) ### 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 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 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 . ...@@ -76,7 +76,7 @@ pip install e .
### Anaconda(方法三) ### Anaconda(方法三)
此处提供本地配置、编译的详细步骤,例如: 此处提供本地配置、编译的详细步骤,例如:
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.sourcefind.cn/tool/)开发者社区下载安装。
``` ```
#DTK驱动:dtk24.04 #DTK驱动:dtk24.04
# python:python3.10 # python:python3.10
...@@ -241,7 +241,7 @@ mamba_ssm (pretrained=state-spaces/mamba2attn-2.7b), gen_kwargs: (None), limit: ...@@ -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 运行推理时会自动连接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 - https://github.com/state-spaces/mamba
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