From 6f9503ab53d007fa828b4677c1e6291a794424b0 Mon Sep 17 00:00:00 2001 From: 51616 Date: Thu, 12 Jun 2025 16:03:06 +0000 Subject: [PATCH] eval shape mismatch fix --- src/ctx_to_lora/data/processing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/ctx_to_lora/data/processing.py b/src/ctx_to_lora/data/processing.py index 21d8826..ff3e7db 100644 --- a/src/ctx_to_lora/data/processing.py +++ b/src/ctx_to_lora/data/processing.py @@ -691,7 +691,7 @@ def construct_and_tokenize_ctx_qa( truncate_middle_if_too_long, fn_kwargs={ "max_length": base_model_max_len, - "columns": ["input_ids", "attention_mask"], + "columns": ["input_ids", "attention_mask", "labels"], "max_new_tokens": 256, }, # batched=True,