From 0f10d2ea04432f9a43a804da0e017f52f84c0a72 Mon Sep 17 00:00:00 2001
From: greyovo <grey030@foxmail.com>
Date: Wed, 20 Dec 2023 19:54:16 +0800
Subject: [PATCH] fix: Error when loading dataset from JSON

---
 finetune_kopa.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/finetune_kopa.py b/finetune_kopa.py
index 1487459..f5ad0ac 100644
--- a/finetune_kopa.py
+++ b/finetune_kopa.py
@@ -179,7 +179,7 @@ def train(
     slama_model = KoPAWithAdapter(model, num_prefix, kge_model=kge_model)
 
     if data_path.endswith(".json") or data_path.endswith(".jsonl"):
-        data = load_dataset("json")
+        data = load_dataset("json", data_files=data_path)
     else:
         data = load_dataset(data_path)