[Doc] Add Baichuan 13B to supported models (#656)

2023-08-02 16:45:12 -07:00
parent 55fe8a81ec
commit f7389f4763
3 changed files with 5 additions and 4 deletions
--- a/vllm/model_executor/models/init.py
+++ b/vllm/model_executor/models/init.py
@@ -1,4 +1,5 @@
-from vllm.model_executor.models.baichuan import BaiChuanForCausalLM, BaichuanForCausalLM
+from vllm.model_executor.models.baichuan import (BaiChuanForCausalLM,
+                                                 BaichuanForCausalLM)
 from vllm.model_executor.models.bloom import BloomForCausalLM
 from vllm.model_executor.models.falcon import FalconForCausalLM
 from vllm.model_executor.models.gpt2 import GPT2LMHeadModel