Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
a7f341c3
Unverified
Commit
a7f341c3
authored
Feb 24, 2026
by
haosdent
Committed by
GitHub
Feb 23, 2026
Browse files
[Bugfix] Fix MRotaryEmbedding missing `truncate` attr with YaRN scaling (#35080)
Signed-off-by:
haosdent
<
haosdent@gmail.com
>
parent
d13ece38
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
0 deletions
+2
-0
vllm/model_executor/layers/rotary_embedding/mrope.py
vllm/model_executor/layers/rotary_embedding/mrope.py
+2
-0
No files found.
vllm/model_executor/layers/rotary_embedding/mrope.py
View file @
a7f341c3
...
@@ -218,12 +218,14 @@ class MRotaryEmbedding(RotaryEmbeddingBase):
...
@@ -218,12 +218,14 @@ class MRotaryEmbedding(RotaryEmbeddingBase):
attn_factor
:
float
=
1
,
attn_factor
:
float
=
1
,
beta_fast
:
int
=
32
,
beta_fast
:
int
=
32
,
beta_slow
:
int
=
1
,
beta_slow
:
int
=
1
,
truncate
:
bool
=
True
,
)
->
None
:
)
->
None
:
self
.
scaling_factor
=
scaling_factor
self
.
scaling_factor
=
scaling_factor
self
.
extrapolation_factor
=
extrapolation_factor
self
.
extrapolation_factor
=
extrapolation_factor
self
.
attn_factor
=
attn_factor
self
.
attn_factor
=
attn_factor
self
.
beta_fast
=
beta_fast
self
.
beta_fast
=
beta_fast
self
.
beta_slow
=
beta_slow
self
.
beta_slow
=
beta_slow
self
.
truncate
=
truncate
if
self
.
scaling_factor
is
not
None
:
if
self
.
scaling_factor
is
not
None
:
# Get n-d magnitude scaling corrected for interpolation
# Get n-d magnitude scaling corrected for interpolation
self
.
mscale
=
float
(
yarn_get_mscale
(
self
.
scaling_factor
)
*
attn_factor
)
self
.
mscale
=
float
(
yarn_get_mscale
(
self
.
scaling_factor
)
*
attn_factor
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment