Fixed gemma example sampler test test_models_from_kaggle - #5477

Fixed gemma example sampler test test_models_from_kaggle#5477

copybara-service[bot] merged 1 commit into

maingoogle/flax:mainfrom

fix-gemma-sampler-test-kagglegoogle/flax:fix-gemma-sampler-test-kaggleCopy head branch name to clipboard