Woosuk Kwon 0fca3cdcf2 [Misc] Enhance attention selector (#4751)

2024-05-13 10:47:25 -07:00

..

__init__.py

[Speculative decoding 2/9] Multi-step worker for draft model (#2424)

2024-01-21 16:31:47 -08:00

test_model_runner.py

[Misc] Enhance attention selector (#4751)

2024-05-13 10:47:25 -07:00

test_swap.py

[Core][Optimization] change python dict to pytorch tensor for blocks to swap (#4659)

2024-05-08 12:07:05 -07:00