423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-23 22:32:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix some params of visual regularize

Browse Source 
Former-commit-id: 1d09d592d3beacca6a17bb3e5a9728150741e551
This commit is contained in:
BUAADreamer 2024-09-29 12:38:25 +08:00
parent 0e33902f61
commit 615edf937e
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