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

xjpang · Mar 4, 2024 · dbcc122 · dbcc122
1 parent fe2b6f3
commit dbcc122
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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()