squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
78 Commits 1 Branch 0 Tags 24 MiB
Python 83.3%
Cuda 11.4%
C++ 3.3%
C 0.8%
Shell 0.7%
Other 0.4%
0deacbce6e
Go to file
Woosuk Kwon 0deacbce6e
Implement single_query_cached_kv_attention kernel (#3)
2023-03-01 15:02:19 -08:00
cacheflow Implement single_query_cached_kv_attention kernel (#3) 2023-03-01 15:02:19 -08:00
csrc Implement single_query_cached_kv_attention kernel (#3) 2023-03-01 15:02:19 -08:00
tests/kernels Implement single_query_cached_kv_attention kernel (#3) 2023-03-01 15:02:19 -08:00
.gitignore Add gitignore 2023-02-16 07:47:21 +00:00
README.md Add README 2023-02-24 12:04:49 +00:00
server.py Clean up the server script 2023-02-24 11:56:21 +00:00
setup.py Implement single_query_cached_kv_attention kernel (#3) 2023-03-01 15:02:19 -08:00

README.md

CacheFlow

Installation

pip install cmake torch transformers
pip install -e .

Run

python server.py