apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

eval shape mismatch fix

Browse source
This commit is contained in:
51616 2025-06-12 16:03:06 +00:00
parent eb87612f91
commit 6f9503ab53
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/ctx_to_lora/data/processing.py
View file

@ -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,