自学内容网 自学内容网

Mamba所需的causal-conv1d 和mamba-ssm库在哪下载?

背景介绍

参照  Mamba [state-spaces/mamba: Mamba SSM architecture (github.com)] github中提到的环境安装[Installation 一栏]

[Option] pip install causal-conv1d>=1.4.0: an efficient implementation of a simple causal Conv1d layer used inside the Mamba block.
pip install mamba-ssm: the core Mamba package.
pip install mamba-ssm[causal-conv1d]: To install core Mamba package and causal-conv1d.
pip install mamba-ssm[dev]: To install core Mamba package and dev depdencies.

在运行 pip install 时, building wheel卡住不动,猜测是网络请求问题,故想要找到 库的地址,先下载到本地。

1 - 【结论】下载地址

causal-conv1d: Releases · Dao-AILab/causal-conv1d (github.com)

mamba-ssm: Releases · state-spaces/mamba (github.com)

2 - 【如何找到的】

参考:Building wheel for opencv-python (pyproject.toml)卡住,解决方案(已解决)-CSDN博客

当遇到 building wheel卡住不动时,(可能是网络请求导致的下载缓慢,抑或是编译的时间太久),可以通过pip命令后增加  --verbose  使得输出更多的调试信息

  • 通常可以查看 网络请求、包安装进度、依赖项、内部决策(安装顺序等)

pip install mamba-ssm -i https://pypi.tuna.tsinghua.edu.cn/simple --verbose 

查看pip install的详细过程。从中就可以找到模型库的download的地址URL:

然后就可以找到地址:https://github.com/state-spaces/mamba/releases /...

 


原文地址:https://blog.csdn.net/m0_46427461/article/details/142413252

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!