nit

d6d49e93 · Baber · 987ac3d6 · d6d49e93
Commit d6d49e93 authored Oct 16, 2025 by Baber
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 2 deletions

lm_eval/models/vllm_causallms.py lm_eval/models/vllm_causallms.py +7 -2

No files found.
--- a/lm_eval/models/vllm_causallms.py
+++ b/lm_eval/models/vllm_causallms.py
@@ -283,11 +283,16 @@ class VLLM(TemplateLM):

    @property
    def max_length(self):
-        return (
+        max_l = (
            8096
-            if self._max_length_internal and self._max_length_internal > 8096
+            if (
+                isinstance(self._max_length_internal, int)
+                and self._max_length_internal > 8096
+            )
            else self._max_length
        )
+        assert isinstance(max_l, int)
+        return max_l

    @property
    def _max_length_internal(self):