ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tiny fix

Browse Source 
Former-commit-id: 3c5da617cdab34c6cae038e3a06d0468ae4c6c86
This commit is contained in:
hiyouga
2023-06-05 15:25:22 +08:00
parent 063a83ab4e
commit a817801c0f
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: