mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-03-30 03:17:19 +08:00
[Fix] Fissertion on YaRN model len (#2984)
This commit is contained in:
parent
57f044945f
commit
f7c1234990
@ -354,7 +354,6 @@ def get_rope(
|
||||
elif scaling_type == "yarn":
|
||||
original_max_position = rope_scaling[
|
||||
"original_max_position_embeddings"]
|
||||
assert max_position == original_max_position * scaling_factor
|
||||
extra_kwargs = {
|
||||
k: v
|
||||
for k, v in rope_scaling.items()
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user