requirements.txt 545 B

1234567891011121314
  1. torch>=2.1.0
  2. torchvision>=0.3
  3. numpy
  4. opencv-python
  5. matplotlib
  6. kornia>=0.6.11
  7. # mambaglue/mambaglue.py 直接依赖(原仓库未写全)
  8. omegaconf
  9. einops
  10. requests
  11. # mamba-ssm:PyPI 无 Windows 官方轮子。用 curl 下载见项目 vendor/mamba-ssm-wheels/*.cmd
  12. # - FuouM(HF,cp311):https://huggingface.co/FuouM/mamba-ssm-windows-builds/tree/main
  13. # - loscrossos(cp312,需匹配 torch/cuda 标签):https://github.com/loscrossos/lib_mamba/releases
  14. # - Linux:pip install mamba-ssm 或 GitHub Releases 的 manylinux 轮子