Skip to content

GitLab

Unverified Commit fe470ae5 authored by Woosuk Kwon's avatar Woosuk Kwon Committed by GitHub
Browse files

[Minor] Fix code style for baichuan (#2003)

parent 3a8c2381
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
vllm/model_executor/models/baichuan.py
View file @ fe470ae5
...@@ -366,8 +366,8 @@ class BaiChuanBaseForCausalLM(nn.Module): ...@@ -366,8 +366,8 @@ class BaiChuanBaseForCausalLM(nn.Module):
weight_loader(param, loaded_weight) weight_loader(param, loaded_weight)
class BaichuanForCausalLM(BaiChuanBaseForCausalLM class BaichuanForCausalLM(BaiChuanBaseForCausalLM):
): # baichuan 13b, baichuan2 13b, baichuan2 7b """Baichuan 13B and Baichuan2 7B/13B."""
def __init__(self, def __init__(self,
config, config,
...@@ -378,7 +378,8 @@ class BaichuanForCausalLM(BaiChuanBaseForCausalLM ...@@ -378,7 +378,8 @@ class BaichuanForCausalLM(BaiChuanBaseForCausalLM
super().__init__(config, "ALIBI", linear_method) super().__init__(config, "ALIBI", linear_method)
class BaiChuanForCausalLM(BaiChuanBaseForCausalLM): # baichuan 7b class BaiChuanForCausalLM(BaiChuanBaseForCausalLM):
"""Baichuan 7B."""
def __init__(self, def __init__(self,
config, config,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment