diff --git a/mindspeed_llm/training/arguments.py b/mindspeed_llm/training/arguments.py index 161ffc841243fd88eed26e3e4ff3b04cbe9b23f2..8cecfd5892b188a00118ad808d76b8d44cf01fa0 100644 --- a/mindspeed_llm/training/arguments.py +++ b/mindspeed_llm/training/arguments.py @@ -258,6 +258,7 @@ def _validate_dualpipe_args(args): DualpipeVFeature().validate_args(args) + def _validate_varlen_fa_args(args): # varlen FA layout must be TND if args.reset_position_ids: