Fix bnb training test failure (#34414)

Commit

1 year ago

Fix bnb training test failure (#34414) * Fix bnb training test: compatibility with OPTSdpaAttention