forked from Karylab-cklius/vllm
d359cda5fae1c9a6fe54ba12b940572edfbf87ac
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>]
Description
A high-throughput and memory-efficient inference and serving engine for LLMs
1.6 GiB
Languages
Python
81.1%
Rust
6.5%
Cuda
4.8%
C++
3.4%
JavaScript
2.7%
Other
1.3%