Ronghang Hu 1034ee2a1a better support for non-CUDA devices (CPU, MPS) (#192)

2024-08-12 10:46:50 -07:00

..

__init__.py

Initial commit

2024-07-29 21:54:20 +00:00

amg.py

Initial commit

2024-07-29 21:54:20 +00:00

misc.py

better support for non-CUDA devices (CPU, MPS) (#192)

2024-08-12 10:46:50 -07:00

transforms.py

Make it optional to build CUDA extension for SAM 2; also fallback to all available kernels if Flash Attention fails (#155)

2024-08-06 10:52:01 -07:00