fix bnb multi gpu training (#2714)

Commit

2 years ago

fix bnb multi gpu training (#2714) * fix bnb multi gpu training * style * elif instead * fix * style * fix