Remove commented-out import and related code for MRotaryEmbedding in patch.py to clean up the codebase.
Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
1 parent
bf51318176
commit
acb28f3d1d
1 file changed
+6
-6
+6
-6
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user