Skip to content

Commit 9259b2c

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

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/models/deepseek_mtp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,9 @@
3636
from vllm.model_executor.models.deepseek_mtp import (
3737
DeepSeekMTP, DeepSeekMultiTokenPredictor, DeepSeekMultiTokenPredictorLayer,
3838
SharedHead)
39-
from vllm.model_executor.models.utils import maybe_prefix
4039
from vllm.model_executor.models.deepseek_v2 import \
4140
get_spec_layer_idx_from_weight_name
41+
from vllm.model_executor.models.utils import maybe_prefix
4242
from vllm.model_executor.sampling_metadata import SamplingMetadata
4343
from vllm.sequence import IntermediateTensors
4444

0 commit comments

Comments
 (0)