1. add model and dataset info to support webui

2026-03-10 13:56:00 +08:00 · 2024-10-10 16:46:34 +08:00
parent 24ebe187e3
commit 0f669f221a
7 changed files with 28 additions and 7 deletions
--- a/src/llamafactory/data/parser.py
+++ b/src/llamafactory/data/parser.py
@@ -98,6 +98,7 @@ def get_dataset_list(dataset_names: Optional[Sequence[str]], dataset_dir: str) -
    dataset_list: List["DatasetAttr"] = []
    for name in dataset_names:
        if dataset_info is None: # dataset_dir is ONLINE
+            load_from = None
            if use_openmind():
                load_from = "om_hub"
            if use_modelscope():