Skip to content

Commit 613b8f1

Browse files
committed
support new processor arg video_maxlen_ttl
1 parent 759de1e commit 613b8f1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/llamafactory/data/mm_plugin.py

+1
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@ def _cal_max_frames_each_video(durations: list, video_maxlen_ttl: int, video_max
122122
)
123123
return max_nums_of_frames
124124

125+
125126
def _get_paligemma_token_type_ids(imglens: list[int], seqlens: list[int], processor: "MMProcessor") -> list[list[int]]:
126127
r"""Get paligemma token type ids for computing loss.
127128

0 commit comments

Comments
 (0)