fix crash in optimizer.step when fsdp2 is enabled and model is bfloat16#3905

SunMarc merged 1 commit intohuggingface:mainhuggingface/accelerate:mainfrom

sywangyi:np_parall_trainer_fixsywangyi/accelerate:np_parall_trainer_fixCopy head branch name to clipboard