Woosuk Kwon b57e6c5949

[Kernel] Add flash-attn back (#4907)

2024-05-19 18:11:30 -07:00

..

__init__.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

test_aqlm.py

AQLM CUDA support (#3287)

2024-04-23 13:59:33 -04:00

test_big_models.py

[Kernel] Add flash-attn back (#4907)

2024-05-19 18:11:30 -07:00

test_embedding.py

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

test_fp8.py

[Kernel] Add flash-attn back (#4907)

2024-05-19 18:11:30 -07:00

test_gptq_marlin_24.py

Add GPTQ Marlin 2:4 sparse structured support (#4790)

2024-05-16 12:56:15 -04:00

test_gptq_marlin.py

[Kernel] add bfloat16 support for gptq marlin kernel (#4788)

2024-05-16 09:55:29 -04:00

test_llava.py

[Test] Make model tests run again and remove --forked from pytest (#3631)

2024-03-28 21:06:40 -07:00

test_marlin.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

test_mistral.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

test_models.py

[Misc]Add customized information for models (#4132)

2024-04-30 21:18:14 -07:00

test_oot_registration.py

[Core] enable out-of-tree model register (#3871)

2024-04-06 17:11:41 -07:00

test_registry.py

[Bugfix][Model] Add base class for vision-language models (#4809)

2024-05-19 00:13:33 -07:00

utils.py

[Kernel] Marlin Expansion: Support AutoGPTQ Models with Marlin (#3922)

2024-04-29 09:35:34 -07:00