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

tiny fix

Browse Source
This commit is contained in:
hiyouga
2023-06-05 15:25:22 +08:00
parent 38b83533a4
commit 44298c1235
3 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/utils/config.py
View File

@@ -148,7 +148,8 @@ class DataTrainingArguments:
def __post_init__(self): # support mixing multiple datasets
dataset_names = [ds.strip() for ds in self.dataset.split(",")]
dataset_info = json.load(open(os.path.join(self.dataset_dir, "dataset_info.json"), "r"))
with open(os.path.join(self.dataset_dir, "dataset_info.json"), "r") as f:
dataset_info = json.load(f)
self.dataset_list: List[DatasetAttr] = []
for name in dataset_names: