Migrate multipack to refactored flash attention

[casper-hansen]

⚠️ Please check that this feature request hasn't been suggested before.

• I searched previous Ideas in Discussions didn't find any similar feature requests.
• I searched previous Issues didn't find any similar feature requests.

🔖 Feature description

Transformers recently refactored flash attention. At current point in time, axolotl replaces the _get_unpad_data for its multipack feature. Instead, replace the central _get_unpad_data in the new transformers.
huggingface/transformers#31446

✔️ Solution

Replace the long if-statement in monkeypatch/multipack and directly replace the centralized unpad function:

        transformers.modeling_flash_attention_utils._get_unpad_data = (  # pylint: disable=protected-access
            get_unpad_data
        )

❓ Alternatives

No response

📝 Additional Context

No response

Acknowledgements

• My issue title is concise, descriptive, and in title casing.
• I have searched the existing issues to make sure this feature has not been requested yet.
• I have provided enough information for the maintainers to understand and evaluate this request.

[winglian]

this should be fixed with #1773 , I just need to do some manual testing