Update vllm/config.py

Co-authored-by: Zhuohan Li <zhuohan123@gmail.com>
This commit is contained in:
Lucas Wilkinson 2025-01-30 08:51:42 -05:00 committed by GitHub
parent 2326814c11
commit 2b140debbb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -970,7 +970,7 @@ class ModelConfig:
return ModelRegistry.is_cross_encoder_model(architectures)
@property
def should_use_mla(self) -> bool:
def use_mla(self) -> bool:
use_mla = (self.is_deepseek_mla and not self.disable_mla
and not envs.VLLM_MLA_DISABLE)
return use_mla