Fix FSDP bnb error (#42600)
* fix
* Apply style fixes
* Fix
* Apply style fixes
* rm
---------
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
