Bug fix in lora.py (#468)

2025-06-25 01:41:19 +08:00 · 2024-02-20 12:53:30 -08:00 · 2024-02-20 12:53:30 -08:00 · 15ecf692b9
commit 15ecf692b9
parent 20b39c7fee
1 changed files with 1 additions and 1 deletions
--- 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