Skip to content

Commit

Permalink
fix: reduce sample_packing warning (#1484)
Browse files Browse the repository at this point in the history
  • Loading branch information
NanoCode012 authored Apr 6, 2024
1 parent bf4cd67 commit bda48f0
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/axolotl/utils/config/models/input/v0_4_1/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -664,8 +664,8 @@ def check_sample_packing_wo_flash(cls, data):
and not data.get("flash_attention")
and not data.get("sdp_attention")
):
raise ValueError(
"sample_packing requires flash_attention or sdp_attention to be set to true"
LOG.warning(
"sample_packing without flash_attention or sdp_attention does not handle cross-attention."
)

return data
Expand Down

0 comments on commit bda48f0

Please sign in to comment.