[Fix] Fissertion on YaRN model len (#2984)

f7c12349 · Woosuk Kwon · GitHub · 57f04494 · f7c12349
Unverified Commit f7c12349 authored Feb 23, 2024 by Woosuk Kwon Committed by GitHub Feb 23, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

vllm/model_executor/layers/rotary_embedding.py vllm/model_executor/layers/rotary_embedding.py +0 -1

No files found.
--- a/vllm/model_executor/layers/rotary_embedding.py
+++ b/vllm/model_executor/layers/rotary_embedding.py
@@ -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()