Woosuk KwonandGitHub 09e9245478 Add custom kernel for RMS normalization (#16)

2023-04-01 00:51:22 +08:00

..

http_frontend

Implement preemption via recomputation & Refactor scheduling logic (#12)

2023-03-30 14:51:46 -07:00

master

Implement preemption via recomputation & Refactor scheduling logic (#12)

2023-03-30 14:51:46 -07:00

models

Add custom kernel for RMS normalization (#16)

2023-04-01 00:51:22 +08:00

parallel_utils

Optimize tensor parallel execution speed (#17)

2023-04-01 00:51:08 +08:00

worker

Support tensor parallel (#2)

2023-03-21 13:45:42 -07:00

block.py

Support beam search & parallel generation (#7)

2023-03-10 09:58:21 -08:00

sampling_params.py

FastAPI-based working frontend (#10)

2023-03-29 14:48:56 +08:00

sequence.py

Implement preemption via recomputation & Refactor scheduling logic (#12)

2023-03-30 14:51:46 -07:00

utils.py

FastAPI-based working frontend (#10)

2023-03-29 14:48:56 +08:00