Skip to content

Commit 421e80d

Browse files
Addthe interface in models executor
Signed-off-by: Divyansh Singhvi <[email protected]>
1 parent 48db105 commit 421e80d

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

vllm/model_executor/models/__init__.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
11
# SPDX-License-Identifier: Apache-2.0
22
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
33

4-
from .interfaces import (HasInnerState, SupportsLoRA, SupportsMultiModal,
5-
SupportsPP, SupportsTranscription, SupportsV0Only,
6-
has_inner_state, supports_lora, supports_multimodal,
7-
supports_pp, supports_transcription, supports_v0_only)
4+
from .interfaces import (HasInnerState, SupportsLoRA, SupportsMRoPE,
5+
SupportsMultiModal, SupportsPP, SupportsTranscription,
6+
SupportsV0Only, has_inner_state, supports_lora,
7+
supports_multimodal, supports_pp,
8+
supports_transcription, supports_v0_only)
89
from .interfaces_base import (VllmModelForPooling, VllmModelForTextGeneration,
910
is_pooling_model, is_text_generation_model)
1011
from .registry import ModelRegistry
@@ -27,4 +28,5 @@
2728
"supports_transcription",
2829
"SupportsV0Only",
2930
"supports_v0_only",
31+
"SupportsMRoPE",
3032
]

0 commit comments

Comments
 (0)