Fix idefics2 llava in the wild bugs

kcz358 · kcz358 · commit 6aeb5504e74e · 2024-05-23T03:10:27.000Z
diff --git a/lmms_eval/models/idefics2.py b/lmms_eval/models/idefics2.py
@@ -196,6 +196,7 @@ def _collate(x):
             gen_kwargs = all_gen_kwargs[0]
             # 
             until = gen_kwargs.pop("until", None)
+            image_aspect_ratio = gen_kwargs.pop("image_aspect_ratio",  None)
             prompts = []
             for context, visual in zip(contexts, visuals):
                 content = []

Original file line number	Diff line number	Diff line change
`@@ -196,6 +196,7 @@ def _collate(x):`
`196`	`196`	`gen_kwargs = all_gen_kwargs[0]`
`197`	`197`	`#`
`198`	`198`	`until = gen_kwargs.pop("until", None)`
	`199`	`+ image_aspect_ratio = gen_kwargs.pop("image_aspect_ratio", None)`
`199`	`200`	`prompts = []`
`200`	`201`	`for context, visual in zip(contexts, visuals):`
`201`	`202`	`content = []`