diff --git a/vllm_ascend/worker/worker_v1.py b/vllm_ascend/worker/worker_v1.py index d8918abf9c..88667a651f 100644 --- a/vllm_ascend/worker/worker_v1.py +++ b/vllm_ascend/worker/worker_v1.py @@ -22,7 +22,6 @@ import torch import torch.nn as nn import torch_npu -from torch_npu.op_plugin.atb._atb_ops import _register_atb_extensions from vllm import envs from vllm.config import VllmConfig from vllm.distributed import (ensure_model_parallel_initialized, @@ -64,7 +63,6 @@ def __init__( # Register ops when worker init. from vllm_ascend import ops ops.register_dummy_fusion_op() - _register_atb_extensions() # init ascend config init_ascend_config(vllm_config)