Skip to content

Commit 766f20b

Browse files
committed
Resolve formatting issues
Signed-off-by: weisirui-eng <weisirui@h-partners.com>
1 parent 3236d32 commit 766f20b

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

vllm_ascend/torchair/mtp_torchair_proposer.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,22 +4,23 @@
44
import torch.nn as nn
55
import torchair
66
import vllm.envs as envs_vllm
7-
from vllm.attention.layer import Attention
87
from torchair import patch_for_hcom
8+
from vllm.attention.layer import Attention
99
from vllm.config import (VllmConfig, get_layers_from_vllm_config,
1010
set_current_vllm_config)
1111
from vllm.forward_context import BatchDescriptor, get_forward_context
12+
from vllm.model_executor.model_loader import get_model_loader
13+
from vllm.model_executor.model_loader.utils import (
14+
process_weights_after_loading, set_default_torch_dtype)
1215
from vllm.v1.core.sched.output import SchedulerOutput
1316
from vllm.v1.sample.metadata import SamplingMetadata
1417
from vllm.v1.spec_decode.metadata import SpecDecodeMetadata
15-
from vllm.model_executor.model_loader.utils import (
16-
process_weights_after_loading, set_default_torch_dtype)
17-
from vllm.model_executor.model_loader import get_model_loader
1818
from vllm_ascend.ascend_config import get_ascend_config
1919
from vllm_ascend.ascend_forward_context import set_ascend_forward_context
2020
from vllm_ascend.attention.utils import AscendCommonAttentionMetadata
2121
from vllm_ascend.spec_decode import MtpProposer
22-
from vllm_ascend.torchair.models.torchair_deepseek_mtp import TorchairDeepSeekMTP
22+
from vllm_ascend.torchair.models.torchair_deepseek_mtp import \
23+
TorchairDeepSeekMTP
2324
from vllm_ascend.torchair.utils import (TORCHAIR_CACHE_DIR,
2425
TorchairCommonAttentionMetadata)
2526
from vllm_ascend.utils import ProfileExecuteDuration, lmhead_tp_enable

0 commit comments

Comments
 (0)