From 27b1d26a6264650861aa3d655435da8b2944cd15 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Sun, 10 Dec 2023 03:10:29 +0000 Subject: [PATCH] [Minor] Fix code style for baichuan --- vllm/model_executor/models/baichuan.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/vllm/model_executor/models/baichuan.py b/vllm/model_executor/models/baichuan.py index f7a3b90a3a02..4a07c3954531 100644 --- a/vllm/model_executor/models/baichuan.py +++ b/vllm/model_executor/models/baichuan.py @@ -366,8 +366,8 @@ def load_weights(self, weight_loader(param, loaded_weight) -class BaichuanForCausalLM(BaiChuanBaseForCausalLM - ): # baichuan 13b, baichuan2 13b, baichuan2 7b +class BaichuanForCausalLM(BaiChuanBaseForCausalLM): + """Baichuan 13B and Baichuan2 7B/13B.""" def __init__(self, config, @@ -378,7 +378,8 @@ def __init__(self, super().__init__(config, "ALIBI", linear_method) -class BaiChuanForCausalLM(BaiChuanBaseForCausalLM): # baichuan 7b +class BaiChuanForCausalLM(BaiChuanBaseForCausalLM): + """Baichuan 7B.""" def __init__(self, config,