fix: reduce sample_packing warning (#1484)

This commit is contained in:
NanoCode012
2024-04-06 21:04:07 +09:00
committed by GitHub
parent bf4cd67252
commit bda48f0150

View File

@@ -664,8 +664,8 @@ class AxolotlInputConfig(
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