From 7ba548856929c2f450cc9ac8bec7efb438e6ace2 Mon Sep 17 00:00:00 2001 From: hiyouga <467089858@qq.com> Date: Tue, 15 Oct 2024 10:10:16 +0800 Subject: [PATCH] fix #5705 Former-commit-id: 3df39f37e1b7988720f36729830889a4c05399b1 --- src/llamafactory/webui/components/eval.py | 2 +- src/llamafactory/webui/components/train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/llamafactory/webui/components/eval.py b/src/llamafactory/webui/components/eval.py index b522913e..8d26d232 100644 --- a/src/llamafactory/webui/components/eval.py +++ b/src/llamafactory/webui/components/eval.py @@ -42,7 +42,7 @@ def create_eval_tab(engine: "Engine") -> Dict[str, "Component"]: elem_dict.update(dict(dataset_dir=dataset_dir, dataset=dataset, **preview_elems)) with gr.Row(): - cutoff_len = gr.Slider(minimum=4, maximum=65536, value=1024, step=1) + cutoff_len = gr.Slider(minimum=4, maximum=131072, value=1024, step=1) max_samples = gr.Textbox(value="100000") batch_size = gr.Slider(minimum=1, maximum=1024, value=2, step=1) predict = gr.Checkbox(value=True) diff --git a/src/llamafactory/webui/components/train.py b/src/llamafactory/webui/components/train.py index e5dc92c3..a167fdeb 100644 --- a/src/llamafactory/webui/components/train.py +++ b/src/llamafactory/webui/components/train.py @@ -68,7 +68,7 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]: ) with gr.Row(): - cutoff_len = gr.Slider(minimum=4, maximum=65536, value=1024, step=1) + cutoff_len = gr.Slider(minimum=4, maximum=131072, value=1024, step=1) batch_size = gr.Slider(minimum=1, maximum=1024, value=2, step=1) gradient_accumulation_steps = gr.Slider(minimum=1, maximum=1024, value=8, step=1) val_size = gr.Slider(minimum=0, maximum=1, value=0, step=0.001)