Skip to content

test_custom_4d_attention_mask skip with sliding window attn#30833

Merged
ArthurZucker merged 1 commit intohuggingface:mainfrom
poedator:4d_test_fix
May 23, 2024
Merged

test_custom_4d_attention_mask skip with sliding window attn#30833
ArthurZucker merged 1 commit intohuggingface:mainfrom
poedator:4d_test_fix

Conversation

@poedator
Copy link
Copy Markdown
Contributor

This is a follow-up to #30348
tests/models/qwen2/test_modeling_qwen2.py::Qwen2ModelTest::test_custom_4d_attention_mask was failing because of use of sliding window attention.
While this sliding window attention does not preclude use of custom 4D masks, it requires modification of the original test_custom_4d_attention_mask.

Summoning @ArthurZucker @gante to review

Copy link
Copy Markdown
Collaborator

@ArthurZucker ArthurZucker left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sounds good, cc @ydshieh will help you 😉

@ArthurZucker ArthurZucker merged commit 6739e1d into huggingface:main May 23, 2024
@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@poedator poedator deleted the 4d_test_fix branch January 14, 2025 13:45
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants