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

Temirulan · Feb 23, 2024 · 914bd11 · 914bd11
1 parent e405220
commit 914bd11
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()