Woosuk Kwon e548c1488a Add support for GPT-2 (#60)

2023-05-04 02:59:56 -07:00

..

http_frontend

New weight loader without np copy (#52)

2023-05-03 15:32:04 +08:00

master

Support bfloat16 data type (#54)

2023-05-03 14:09:44 -07:00

models

Add support for GPT-2 (#60)

2023-05-04 02:59:56 -07:00

parallel_utils

Add CUDA graph-based all reduce launcher (#26)

2023-04-05 11:16:57 -07:00

worker

New weight loader without np copy (#52)

2023-05-03 15:32:04 +08: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

Collect system stats in scheduler & Add scripts for experiments (#30)

2023-04-12 15:03:49 -07:00

utils.py

FastAPI-based working frontend (#10)

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