Woosuk Kwon c9d5b6d4a8 Replace FlashAttention with xformers (#70)

2023-05-05 02:01:08 -07:00

..

http_frontend

New weight loader without np copy (#52)

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

master

Replace FlashAttention with xformers (#70)

2023-05-05 02:01:08 -07:00

models

Replace FlashAttention with xformers (#70)

2023-05-05 02:01:08 -07:00

parallel_utils

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

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

worker

Replace FlashAttention with xformers (#70)

2023-05-05 02:01:08 -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