From a63144c28f86e1aea2823b9a0f3137f9c9faee99 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: 0c85fd253f860eee3c7b9b5a4e77ffbf93af372a --- 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)