github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-02 09:38:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Feat]Inference RPC Server Support (#5705)

Browse Source 
* rpc support source
* kv cache logical/physical disaggregation
* sampler refactor
* colossalai launch built in
* Unitest
* Rpyc support

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Runyu Lu
2024-05-14 10:00:55 +08:00
committed by GitHub
parent de4bf3dedf
commit 18d67d0e8e
15 changed files with 1032 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
requirements/requirements-test.txt
View File

@@ -19,4 +19,5 @@ datasets
pydantic
ray
peft>=0.7.1
rpyc==6.0.0
#auto-gptq now not support torch1.12

1
requirements/requirements.txt
View File

@@ -19,3 +19,4 @@ protobuf
transformers==4.36.2
peft>=0.7.1
bitsandbytes>=0.39.0
rpyc==6.0.0