[Minor] Fix max new tokens error in video llava (#237)

EvolvingLMMs-Lab · Sep 10, 2024 · 1618dc7 · 1618dc7
1 parent 35d221f
commit 1618dc7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmms_eval/models/video_llava.py b/lmms_eval/models/video_llava.py
@@ -203,7 +203,7 @@ def generate_until(self, requests) -> List[str]:
             if "num_beams" not in gen_kwargs:
                 gen_kwargs["num_beams"] = 1
 
-            generate_ids = self.model.generate(**inputs, max_length=gen_kwargs["max_new_tokens"], temperature=gen_kwargs["temperature"])
+            generate_ids = self.model.generate(**inputs, max_new_tokens=gen_kwargs["max_new_tokens"], temperature=gen_kwargs["temperature"])
 
             outputs = self._processor.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0].split("ASSISTANT:")[-1].strip()
             res.append(outputs)