From fa7355404c12811c17445d0175a882f4911710bf Mon Sep 17 00:00:00 2001 From: bursteratom Date: Sun, 2 Feb 2025 00:38:35 -0500 Subject: [PATCH] test --- src/axolotl/utils/collators/batching.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/collators/batching.py b/src/axolotl/utils/collators/batching.py index 5a514b613..b68a31d71 100644 --- a/src/axolotl/utils/collators/batching.py +++ b/src/axolotl/utils/collators/batching.py @@ -182,7 +182,7 @@ class FlexBatchSamplerDataCollatorForSeq2Seq(DataCollatorForSeq2Seq): collated_seq_lens, max_seq_len = get_seqlens_from_pos_ids(out["position_ids"]) # out["attention_mask"] = packed_block_causal_mask(collated_seq_lens) out["attention_mask"] = create_block_causal_mask(collated_seq_lens, max_seq_len) - + raise ValueError(f"{out['attention_mask'].shape}") return out