From acb28f3d1d7d277e3f772e0ec2b8b25d5f541d06 Mon Sep 17 00:00:00 2001 From: tzhouam Date: Wed, 28 Jan 2026 16:52:10 +0800 Subject: [PATCH] Remove commented-out import and related code for MRotaryEmbedding in patch.py to clean up the codebase. Signed-off-by: tzhouam --- vllm_omni/patch.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/vllm_omni/patch.py b/vllm_omni/patch.py index 687ff5186..9fb4249e6 100644 --- a/vllm_omni/patch.py +++ b/vllm_omni/patch.py @@ -1,9 +1,10 @@ import sys from vllm.inputs.data import TokensPrompt as _OriginalTokensPrompt -from vllm.model_executor.layers.rotary_embedding import ( - MRotaryEmbedding as _OriginalMRotaryEmbedding, -) + +# from vllm.model_executor.layers.rotary_embedding import ( +# MRotaryEmbedding as _OriginalMRotaryEmbedding, +# ) from vllm.v1.engine import EngineCoreOutput as _OriginalEngineCoreOutput from vllm.v1.engine import EngineCoreOutputs as _OriginalEngineCoreOutputs from vllm.v1.engine import EngineCoreRequest as _OriginalEngineCoreRequest @@ -12,7 +13,6 @@ from vllm.v1.request import Request as _OriginalRequest import vllm_omni.logger # noqa: F401 from vllm_omni.engine import OmniEngineCoreOutput, OmniEngineCoreOutputs, OmniEngineCoreRequest from vllm_omni.inputs.data import OmniTokensPrompt -from vllm_omni.model_executor.layers.mrope import MRotaryEmbedding from vllm_omni.request import OmniRequest for module_name, module in sys.modules.items(): @@ -25,8 +25,8 @@ for module_name, module in sys.modules.items(): module.EngineCoreOutputs = OmniEngineCoreOutputs if hasattr(module, "TokensPrompt") and module.TokensPrompt == _OriginalTokensPrompt: module.TokensPrompt = OmniTokensPrompt - if hasattr(module, "MRotaryEmbedding") and module.MRotaryEmbedding == _OriginalMRotaryEmbedding: - module.MRotaryEmbedding = MRotaryEmbedding + # if hasattr(module, "MRotaryEmbedding") and module.MRotaryEmbedding == _OriginalMRotaryEmbedding: + # module.MRotaryEmbedding = MRotaryEmbedding if hasattr(module, "Request") and module.Request == _OriginalRequest: module.Request = OmniRequest if hasattr(module, "EngineCoreRequest") and module.EngineCoreRequest == _OriginalEngineCoreRequest: