Add support for baichuan (#365)

2023-07-17 21:50:55 +01:00
parent 2bdea7ac11
commit 20b0d88d16
6 changed files with 361 additions and 0 deletions
--- a/vllm/transformers_utils/configs/init.py
+++ b/vllm/transformers_utils/configs/init.py
@@ -1,5 +1,7 @@
 from vllm.transformers_utils.configs.mpt import MPTConfig
+from vllm.transformers_utils.configs.baichuan import BaiChuanConfig

 __all__ = [
    "MPTConfig",
+    "BaiChuanConfig",
 ]