transformers

9e4199ed - Gemma3 fixes (#41572)

Gemma3 fixes (#41572) * Multiple device error fix * FA2 equivalence fix * Move the train fwd in cfg test * Style * Added comment * Made the comment more clear