Skip to content

Fix Llava tests for torch too!#44476

Merged
Rocketknight1 merged 3 commits intomainfrom
fix_llava_test_round_2
Mar 11, 2026
Merged

Fix Llava tests for torch too!#44476
Rocketknight1 merged 3 commits intomainfrom
fix_llava_test_round_2

Conversation

@Rocketknight1
Copy link
Member

I made an oversight in the fix at #43981 - I didn't realize the dim order changed for torch, so the test was still flaky for torch tensors. The fix reduced the flaky frequency a lot so I thought it had been fixed, but actually it's still there, just failing around 0.25% of the time.

I tested this PR with 1000 runs and they all passed, so hopefully this is the end of it! cc @zucchini-nlp

@Rocketknight1 Rocketknight1 marked this pull request as ready for review March 5, 2026 16:39
Copy link
Member

@zucchini-nlp zucchini-nlp left a comment

Choose a reason for hiding this comment

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

Ah right, thanks for catching it

@github-actions
Copy link
Contributor

[For maintainers] Suggested jobs to run (before merge)

run-slow: llava

@HuggingFaceDocBuilderDev

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.

@Rocketknight1 Rocketknight1 added this pull request to the merge queue Mar 11, 2026
Merged via the queue into main with commit 0a9fdea Mar 11, 2026
19 checks passed
@Rocketknight1 Rocketknight1 deleted the fix_llava_test_round_2 branch March 11, 2026 16:47
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