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