[NeMo Gemma] ModuleNotFoundError: No module named 'params'

Hi,
I am trying to convert gemma flax model to .nemo.

The script I ran is based on:
(URL: Checkpoint Conversion - NVIDIA Docs)
pip install orbax jax flax jaxlib;
export PYTHONPATH=/path/to/google/gemma_jax:$PYTHONPATH;
python3 /opt/NeMo/scripts/nlp_language_modeling/convert_gemma_jax_to_nemo.py
–input_name_or_path /path/to/gemma/checkpoints/jax/7b
–output_path /path/to/gemma-7b.nemo
–tokenizer_path /path/to/tokenizer.model

But the error message says “ModuleNotFoundError: No module named ‘params’”
though I cloned deepmind/gemma repo and set $PYTHONPATH variable.
(gemma/gemma/transformer.py at a24194737dcb54b7392091e9ba772aea1cb68ffb · google-deepmind/gemma · GitHub)

Could I get any advice on this??
Thanks.