423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 11:50:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

add dataset stage and filter dataset when stage chosen in webui

Browse Source 
Former-commit-id: c0e4d1e81b
This commit is contained in:
codemayq
2023-08-23 18:54:23 +08:00
parent 2accde2cb7
commit 4b29d9d2b0
4 changed files with 84 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/llmtuner/webui/components/train.py
View File

@@ -22,7 +22,8 @@ def create_train_tab(top_elems: Dict[str, "Component"], runner: "Runner") -> Dic
preview_box, preview_count, preview_samples, close_btn = create_preview_box()
dataset_dir.change(list_dataset, [dataset_dir], [dataset])
training_stage.change(list_dataset, [dataset_dir, training_stage], [dataset])
dataset_dir.change(list_dataset, [dataset_dir, training_stage], [dataset])
dataset.change(can_preview, [dataset_dir, dataset], [data_preview_btn])
data_preview_btn.click(
get_preview,