|
@@ -87,7 +87,7 @@ class ModelTrainer:
|
|
data.append(json.loads(line))
|
|
data.append(json.loads(line))
|
|
|
|
|
|
# 将列表转换为 HuggingFace Dataset 对象
|
|
# 将列表转换为 HuggingFace Dataset 对象
|
|
- data = Dataset.from_list(data)
|
|
|
|
|
|
+ train_dataset = Dataset.from_list(data)
|
|
|
|
|
|
return train_dataset
|
|
return train_dataset
|
|
|
|
|