diff --git a/configs/server_config.py.example b/configs/server_config.py.example index 1268787..801e6d2 100644 --- a/configs/server_config.py.example +++ b/configs/server_config.py.example @@ -112,6 +112,9 @@ FSCHAT_MODEL_WORKERS = { "qwen-api": { "port": 21006, }, + "baichuan-api": { + "port": 21007, + }, } # fastchat multi model worker server diff --git a/server/model_workers/__init__.py b/server/model_workers/__init__.py index c1a824b..10f6da1 100644 --- a/server/model_workers/__init__.py +++ b/server/model_workers/__init__.py @@ -4,3 +4,4 @@ from .xinghuo import XingHuoWorker from .qianfan import QianFanWorker from .fangzhou import FangZhouWorker from .qwen import QwenWorker +from .baichuan import BaiChuanWorker