From 15ecf692b915aac5f73f6f9c431a12a0ad4ebbd9 Mon Sep 17 00:00:00 2001 From: Sergey Shumov Date: Tue, 20 Feb 2024 12:53:30 -0800 Subject: [PATCH] Bug fix in lora.py (#468) --- lora/lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lora/lora.py b/lora/lora.py index 64e8d671..f21f0d2d 100644 --- a/lora/lora.py +++ b/lora/lora.py @@ -141,7 +141,7 @@ def load(args): def load_and_check(name): dataset_path = Path(args.data) / f"{name}.jsonl" try: - train = Dataset(dataset_path) + return Dataset(dataset_path) except Exception as e: print(f"Unable to build dataset {dataset_path} ({e})") raise