Skip to content

Commit 251812d

Browse files
author
wangzaijun
committed
fix
1 parent 9a191be commit 251812d

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

lightllm/utils/config_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def _get_config_llm_keyvalue(model_path: str, key_name: str):
2828
if value is None:
2929
logger.error(f"cannot get {key_name} from config.json, return None")
3030

31-
return None
31+
return value
3232

3333

3434
def get_hidden_size(model_path: str) -> Optional[int]:

lightllm/utils/kv_cache_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ def calcu_cpu_cache_meta() -> "CpuKVCacheMeta":
8484
args = get_env_start_args()
8585
assert args.enable_cpu_cache
8686

87-
if get_model_type() not in ["deepseek_v2", "deepseek_v3"]:
87+
if get_model_type(model_path=args.model_dir) not in ["deepseek_v2", "deepseek_v3"]:
8888
item_size = 2
8989
num_key_value_heads = get_num_key_value_heads(args.model_dir) * 2
9090
head_dim = get_head_dim(args.model_dir)

0 commit comments

Comments
 (0)