From 1d09d592d3beacca6a17bb3e5a9728150741e551 Mon Sep 17 00:00:00 2001
From: BUAADreamer <1428195643@qq.com>
Date: Sun, 29 Sep 2024 12:38:25 +0800
Subject: [PATCH] fix some params of visual regularize

---
 src/llamafactory/data/mm_plugin.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/llamafactory/data/mm_plugin.py b/src/llamafactory/data/mm_plugin.py
index bb008ae8..553aef96 100644
--- a/src/llamafactory/data/mm_plugin.py
+++ b/src/llamafactory/data/mm_plugin.py
@@ -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