Skip to content

Commit 7646e00

Browse files
authored
Merge pull request vllm-project#4 from beagleski/bapatra/patching-for-su
patching for having type su
2 parents 561d5a8 + bfd3c80 commit 7646e00

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

vllm/model_executor/layers/rotary_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -513,7 +513,7 @@ def get_rope(
513513
base, is_neox_style,
514514
scaling_factor,
515515
**extra_kwargs)
516-
elif scaling_type == 'longrope':
516+
elif scaling_type in ("longrope", "su"):
517517
short_factor = rope_scaling["short_factor"]
518518
long_factor = rope_scaling["long_factor"]
519519
original_max_position = rope_scaling[

0 commit comments

Comments
 (0)