p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add check

This commit is contained in:
codingma 2024-04-16 10:56:39 +08:00
parent 75aa6392e8
commit 62294289dc
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/llmtuner/webui/components/data.py
View File

@ -28,14 +28,13 @@ def can_preview(dataset_dir: str, dataset: list) -> "gr.Button":
dataset_info = json.load(f)
except Exception:
return gr.Button(interactive=False)
if len(dataset) == 0 or "file_name" not in dataset_info[dataset[0]]:
return gr.Button(interactive=False)
local_path = os.path.join(dataset_dir, dataset_info[dataset[0]]["file_name"])
if (
len(dataset) > 0
and "file_name" in dataset_info[dataset[0]]
and (os.path.isfile(local_path)
or (os.path.isdir(local_path)) and len(os.listdir(local_path)) != 0)
):
if (os.path.isfile(local_path)
or (os.path.isdir(local_path) and len(os.listdir(local_path)) != 0)):
return gr.Button(interactive=True)
else:
return gr.Button(interactive=False)