Skip to content

Commit 65a3c5d

Browse files
clean code
Signed-off-by: shikang-hangzhou <459956190@qq.com>
1 parent 812087b commit 65a3c5d

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

vllm_ascend/models/deepseek_dbo.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -33,17 +33,15 @@
3333
from torch import nn
3434
from transformers import PretrainedConfig
3535
from vllm.attention import AttentionMetadata
36-
from vllm.config import (CacheConfig, ModelConfig, VllmConfig,
37-
get_current_vllm_config)
36+
from vllm.config import (CacheConfig, ModelConfig, VllmConfig)
3837
from vllm.distributed import (get_ep_group, get_pp_group,
3938
get_tensor_model_parallel_rank,
4039
get_tensor_model_parallel_world_size,
4140
get_tp_group, tensor_model_parallel_all_reduce)
4241
from vllm.distributed.parallel_state import get_dp_group
4342
from vllm.forward_context import get_forward_context
4443
from vllm.model_executor.layers.layernorm import RMSNorm
45-
from vllm.model_executor.layers.linear import (ReplicatedLinear,
46-
UnquantizedLinearMethod)
44+
from vllm.model_executor.layers.linear import UnquantizedLinearMethod
4745
from vllm.model_executor.layers.logits_processor import LogitsProcessor
4846
from vllm.model_executor.layers.quantization import QuantizationConfig
4947
from vllm.model_executor.layers.sampler import get_sampler
@@ -57,10 +55,8 @@
5755
from vllm.sequence import IntermediateTensors
5856

5957
import vllm_ascend.envs as envs_ascend
60-
from vllm_ascend.ascend_config import get_ascend_config
6158
from vllm_ascend.ascend_forward_context import FusedMoEState
6259
from vllm_ascend.models.deepseek_v2 import (CustomDeepseekV2DecoderLayer,
63-
CustomDeepseekV2MLAAttention,
6460
CustomDeepseekV2MLP,
6561
CustomDeepseekV2MoE)
6662
from vllm_ascend.multistream.base import MSEventKey
@@ -72,7 +68,6 @@
7268
from vllm_ascend.multistream.metadata import (MultiStreamConfig,
7369
MultiStreamStepMetadata,
7470
make_multistream_metadata_ds)
75-
from vllm_ascend.ops.fused_moe import AscendFusedMoE
7671
from vllm_ascend.quantization.w8a8_dynamic import (
7772
AscendW8A8DynamicLinearMethod, apply_mlp)
7873
from vllm_ascend.utils import dispose_tensor

0 commit comments

Comments
 (0)