NameError found flax repo due to case sensitivity

[KumarGitesh2024]

when I was executing fine tuning tutorial notebook I found the code error in this notebook : https://github.com/google-deepmind/gemma/blob/main/colabs/fine_tuning_tutorial.ipynb

And also getting the same error in the below notebooks, could you please recheck it
https://github.com/google-deepmind/gemma/blob/main/colabs/gsm8k_eval.ipynb
https://github.com/google-deepmind/gemma/blob/main/colabs/sampling_tutorial.ipynb
I am providing a gist for reference .

[Gopi-Uppari]

Hi @KumarGitesh2024,

I was able to reproduce the issue, and to prevent this error, it's necessary to change the variant to VARIANT. Additionally, some users have already raised pull requests to address this problem. Our team will review and provide feedback as soon as possible.

Thank you.

[Gopi-Uppari]

Hi @KumarGitesh2024,

Could you please confirm if this issue is resolved for you with the above comment ? Please feel free to close the issue if it is resolved ?

Thank you.

[tilakrayal]

Closing this issue as stale. Feel free to re-open if the issue still persists. Thank you!