423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-23 06:12:50 +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,
image_resolution=128,
image_resolution=168,
video_fps=1.0,
video_maxlen=64,
video_maxlen=16,
)
return videos
@ -644,9 +644,9 @@ class VideoLlavaPlugin(BasePlugin):
"""
videos = super()._regularize_videos(
videos,
image_resolution=128,
image_resolution=224,
video_fps=1.0,
video_maxlen=64,
video_maxlen=8,
)
return videos