Skip to content

Commit 252ddb2

Browse files
committed
fix deepseek func
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
1 parent 706348d commit 252ddb2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm_ascend/worker/model_runner_v1.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2420,8 +2420,8 @@ def initialize_kv_cache_tensors(
24202420
if layer_name in self.runner_only_attn_layers:
24212421
continue
24222422
layer_names.add(layer_name)
2423-
assert layer_names == set(kv_cache_raw_tensors.keys(
2424-
)), "Some layers are not correctly initialized"
2423+
# assert layer_names == set(kv_cache_raw_tensors.keys(
2424+
# )), "Some layers are not correctly initialized"
24252425

24262426
def align_memory(tensor: torch.Tensor, alignment: int) -> torch.Tensor:
24272427
data_ptr = tensor.data_ptr()

0 commit comments

Comments
 (0)