Skip to content

Commit daddc7d

Browse files
committed
update ernie45_vl_rope
Signed-off-by: Yan Ma <yan.ma@intel.com>
1 parent 0b92158 commit daddc7d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/model_executor/layers/rotary_embedding/ernie45_vl_rope.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,10 @@
55
import torch
66

77
from .common import apply_rotary_emb_dispatch
8-
from .mrope import MRotaryEmbedding
8+
from .mrope import MRotaryEmbeddingBase
99

1010

11-
class Ernie4_5_VLRotaryEmbedding(MRotaryEmbedding):
11+
class Ernie4_5_VLRotaryEmbedding(MRotaryEmbeddingBase):
1212
"""3D rotary positional embedding. 3D is t:time h:height w:width"""
1313

1414
def forward_native( # type: ignore[override]

0 commit comments

Comments
 (0)