Ronghang Hu 1034ee2a1a better support for non-CUDA devices (CPU, MPS) (#192) 1 년 전
..
__init__.py 0c5f8c5432 Initial commit 1 년 전
amg.py 0c5f8c5432 Initial commit 1 년 전
misc.py 1034ee2a1a better support for non-CUDA devices (CPU, MPS) (#192) 1 년 전
transforms.py 6f7e700c37 Make it optional to build CUDA extension for SAM 2; also fallback to all available kernels if Flash Attention fails (#155) 1 년 전