Skip to content

Commit bced048

Browse files
author
赵江江
committed
feat: add mtp ut case
Signed-off-by: 赵江江 <[email protected]>
1 parent cdcdc3b commit bced048

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm_ascend/models/deepseek_mtp.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,8 @@
3737
DeepSeekMTP, DeepSeekMultiTokenPredictor, DeepSeekMultiTokenPredictorLayer,
3838
SharedHead)
3939
from vllm.model_executor.models.utils import maybe_prefix
40-
from vllm.model_executor.models.deepseek_v2 import get_spec_layer_idx_from_weight_name
40+
from vllm.model_executor.models.deepseek_v2 import \
41+
get_spec_layer_idx_from_weight_name
4142
from vllm.model_executor.sampling_metadata import SamplingMetadata
4243
from vllm.sequence import IntermediateTensors
4344

0 commit comments

Comments
 (0)