| cacheflow | ||
| csrc | ||
| tests/kernels | ||
| .gitignore | ||
| README.md | ||
| server.py | ||
| setup.py | ||
CacheFlow
Installation
pip install psutil numpy torch transformers
pip install flash-attn # This may take up to 10 mins.
pip install -e .
Run
ray start --head
python server.py [--tensor-parallel-size <N>]