lucas
/
LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix some params of visual regularize

This commit is contained in:
BUAADreamer 2024-09-29 12:38:25 +08:00
parent 7f3f81009e
commit 1d09d592d3
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/llamafactory/data/mm_plugin.py
View File

@ -405,9 +405,9 @@ class LlavaNextVideoPlugin(BasePlugin):
""" """
videos = super()._regularize_videos( videos = super()._regularize_videos(
videos, videos,
image_resolution=128, image_resolution=168,
video_fps=1.0, video_fps=1.0,
video_maxlen=64, video_maxlen=16,
) )
return videos return videos
@ -644,9 +644,9 @@ class VideoLlavaPlugin(BasePlugin):
""" """
videos = super()._regularize_videos( videos = super()._regularize_videos(
videos, videos,
image_resolution=128, image_resolution=224,
video_fps=1.0, video_fps=1.0,
video_maxlen=64, video_maxlen=8,
) )
return videos return videos