change to assert

Signed-off-by: Sugar-zsg <[email protected]>

Author: Sugar-zsg

vllm/attention/layers/cross_attention.py

@@ -25,12 +25,10 @@
 def _get_max_encoder_len(vllm_config: "VllmConfig") -> int:
     """Gets the max number of encoder input tokens from the config.
     """
-    max_len = vllm_config.scheduler_config.max_num_encoder_input_tokens
-
-    if isinstance(max_len, int) and max_len > 0:
-        return max_len
-
-    return 0
+    sc = vllm_config.scheduler_config
+    assert sc and isinstance(sc.max_num_encoder_input_tokens, int), \
+        "max_num_encoder_input_tokens must be int for enc-dec models"
+    return sc.max_num_encoder_input_tokens
 
 
 def _get_cross_slot_mapping(encoder_seq_lens: np.ndarray,