Stephen Krider 1356df53bd [Kernel] Use flash-attn for decoding (#3648) ... Co-authored-by: Woosuk Kwon <woosuk.kwon@berkeley.edu> Co-authored-by: LiuXiaoxuanPKU <lilyliupku@gmail.com>		2024-05-13 15:50:33 -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] Use flash-attn for decoding (#3648)	2024-05-13 15:50:33 -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] Use flash-attn for decoding (#3648)	2024-05-13 15:50:33 -07:00
test_gptq_marlin.py	[CI/Build] Move test_utils.py to tests/utils.py (#4425)	2024-05-13 23:50:09 +09: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
utils.py	[Kernel] Marlin Expansion: Support AutoGPTQ Models with Marlin (#3922)	2024-04-29 09:35:34 -07:00