Skip to content

Commit c15ebb3

Browse files
authored
fix: add max_frames_num to OpenAICompatible (#740)
1 parent 8ada805 commit c15ebb3

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

lmms_eval/models/openai_compatible.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ def __init__(
3838
continual_mode: bool = False,
3939
response_persistent_folder: str = None,
4040
azure_openai: bool = False,
41+
max_frames_num: int = 10,
4142
**kwargs,
4243
) -> None:
4344
super().__init__()
@@ -46,6 +47,7 @@ def __init__(
4647
self.max_retries = max_retries
4748
self.max_size_in_mb = max_size_in_mb # some models have a limit on the size of the image
4849
self.continual_mode = continual_mode
50+
self.max_frames_num = max_frames_num
4951
if self.continual_mode:
5052
if response_persistent_folder is None:
5153
raise ValueError("Continual mode requires a persistent path for the response. Please provide a valid path.")

0 commit comments

Comments
 (0)