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

add check

Browse Source 
Former-commit-id: 62294289dc6f594f37c5c6b73585c2403248cc55
This commit is contained in:
codingma 2024-04-16 10:56:39 +08:00
parent 43d1730aa5
commit 64f96cfd6d
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) dataset_info = json.load(f)
except Exception: except Exception:
return gr.Button(interactive=False) 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"]) local_path = os.path.join(dataset_dir, dataset_info[dataset[0]]["file_name"])
if ( if (os.path.isfile(local_path)
len(dataset) > 0 or (os.path.isdir(local_path) and len(os.listdir(local_path)) != 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)
):
return gr.Button(interactive=True) return gr.Button(interactive=True)
else: else:
return gr.Button(interactive=False) return gr.Button(interactive=False)