update max_length

2026-02-04 17:59:18 +08:00 · 2024-08-15 11:05:14 +08:00
parent 8befa86b5e
commit 6f2e885144
1 changed files with 1 additions and 1 deletions
--- 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