Fix intermediates for Gemma example#5409

copybara-service[bot] merged 2 commits intomaingoogle/flax:mainfrom

gemma-example-using-grain--fix-intermediatesgoogle/flax:gemma-example-using-grain--fix-intermediatesCopy head branch name to clipboard