fix gemma3 grad acc (#37208)
* fix gemma3 grad acc
* fix
* fix
* fix
* fix
* rmv print
* rm
* Update setup.py
* Apply style fixes
* propagate the changes
---------
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Arthur <arthur.zucker@gmail.com>
