[Misc] Avoid misleading warning messages (#10438)

Signed-off-by: Jee Jee Li <pandaleefree@gmail.com>
This commit is contained in:
Jee Jee Li
2024-11-19 16:54:58 +08:00
committed by GitHub
parent 272e31c0bd
commit 382b6a4852
2 changed files with 7 additions and 8 deletions

View File

@@ -575,8 +575,7 @@ class ChatGLMModel(nn.Module):
return hidden_states
class ChatGLMBaseModel(nn.Module, SupportsLoRA, SupportsPP,
SupportsMultiModal):
class ChatGLMBaseModel(nn.Module, SupportsLoRA, SupportsPP):
def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
super().__init__()
@@ -695,7 +694,7 @@ class ChatGLM(ChatGLMBaseModel):
embedding_padding_modules = []
class ChatGLMV(ChatGLMBaseModel):
class ChatGLMV(ChatGLMBaseModel, SupportsMultiModal):
packed_modules_mapping = {
"query_key_value": ["query_key_value"],
"dense_h_to_4h": ["dense_h_to_4h"],