mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-14 06:14:34 +08:00
Update vllm/config.py
Co-authored-by: Zhuohan Li <zhuohan123@gmail.com>
This commit is contained in:
parent
2326814c11
commit
2b140debbb
@ -970,7 +970,7 @@ class ModelConfig:
|
|||||||
return ModelRegistry.is_cross_encoder_model(architectures)
|
return ModelRegistry.is_cross_encoder_model(architectures)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def should_use_mla(self) -> bool:
|
def use_mla(self) -> bool:
|
||||||
use_mla = (self.is_deepseek_mla and not self.disable_mla
|
use_mla = (self.is_deepseek_mla and not self.disable_mla
|
||||||
and not envs.VLLM_MLA_DISABLE)
|
and not envs.VLLM_MLA_DISABLE)
|
||||||
return use_mla
|
return use_mla
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user