PR #34414 Fix bnb training test failure

Fix bnb training test failure #34414

matthewdouglas merged 2 commits into main from fix-test-bnb-opt-sdpa

Fix bnb training test: compatibility with OPTSdpaAttention

034c5325

Revert unrelated changes

e55e2e68

matthewdouglas requested a review from

BenjaminBossan 1 year ago

matthewdouglas added Tests

BenjaminBossan approved these changes on 2024-10-25

matthewdouglas changed the title ~~Fix te~~ Fix bnb training test failure 1 year ago

matthewdouglas merged e447185b into main 1 year ago

matthewdouglas deleted the fix-test-bnb-opt-sdpa branch 1 year ago

Reviewers

BenjaminBossan

Assignees

No one assigned

Labels

Tests

Milestone

No milestone