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

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

..

backbones

Update hieradet.py

2024-08-09 11:14:11 +05:30

sam

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

__init__.py

Initial commit

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

memory_attention.py

Initial commit

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

memory_encoder.py

Initial commit

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

position_encoding.py

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

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

sam2_base.py

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

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

sam2_utils.py

Initial commit

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