Woosuk KwonandGitHub a96d63c21d Add support for GPT-NeoX (Pythia) (#50)

2023-04-28 00:32:10 -07:00

..

http_frontend

Add an option to use dummy model weights (#33)

2023-04-08 23:36:12 -07:00

master

Support various block sizes & Change default block size to 16 (#38)

2023-04-15 09:03:24 -07:00

models

Add support for GPT-NeoX (Pythia) (#50)

2023-04-28 00:32:10 -07:00

parallel_utils

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

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

worker

Add an option to use dummy model weights (#33)

2023-04-08 23:36:12 -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

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