From 6f2e8851448377225c27874b6da36c69219ff0ff Mon Sep 17 00:00:00 2001
From: qianyu chen <38046403+qyc-98@users.noreply.github.com>
Date: Thu, 15 Aug 2024 11:05:14 +0800
Subject: [PATCH] update max_length

---
 finetune/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/finetune/dataset.py b/finetune/dataset.py
index d03a325..dbf6bbc 100644
--- a/finetune/dataset.py
+++ b/finetune/dataset.py
@@ -317,7 +317,7 @@ def preprocess(
     llm_type=None,
     patch_size=14,
     batch_vision=False,
-    max_length=None,
+    max_length=2048,
 ):
     """
     single(multi) image(s) preprocess, the image(s) will be placed at the top of the conversation