Cyrus LeungandRoger Wang 5ae5ed1e60 [Core] Consolidate prompt arguments to LLM engines (#4328) ...

Co-authored-by: Roger Wang <ywang@roblox.com>

2024-05-28 13:29:31 -07:00

..

block

[Core] Sliding window for block manager v2 (#4545)

2024-05-28 11:07:07 +09:00

__init__.py

[Tests] Add block manager and scheduler tests (#3108)

2024-03-05 18:23:34 -08:00

test_block_manager.py

[Core] Consolidate prompt arguments to LLM engines (#4328)

2024-05-28 13:29:31 -07:00

test_chunked_prefill_scheduler.py

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

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

test_scheduler.py

[Scheduler] Warning upon preemption and Swapping (#4647)

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

utils.py

[Core] Consolidate prompt arguments to LLM engines (#4328)

2024-05-28 13:29:31 -07:00