Error with `model = gm.nn.Gemma4_E2B()`

[syrkis]

On the main branch of gemma. Getting this:

[Running: uv run python main.py]
Traceback (most recent call last):
  File "/home/nobr/prism/main.py", line 15, in <module>
    model = gm.nn.Gemma4_E2B()
            ^^^^^^^^^^^^^^^^
  File "/home/nobr/prism/.venv/lib/python3.12/site-packages/etils/epy/lazy_api_imports_utils.py", line 119, in _getattr
    symbol = _import_symbol(
             ^^^^^^^^^^^^^^^
  File "/home/nobr/prism/.venv/lib/python3.12/site-packages/etils/epy/lazy_api_imports_utils.py", line 160, in _import_symbol
    module = __import__(module_name, fromlist=[obj_name])
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/nobr/prism/.venv/lib/python3.12/site-packages/gemma/gm/nn/gemma4/_gemma4.py", line 21, in <module>
    from gemma.gm.nn.gemma4 import _config
  File "/home/nobr/prism/.venv/lib/python3.12/site-packages/gemma/gm/nn/gemma4/_config.py", line 24, in <module>
    from gemma.gm.text import _tokenizer
  File "/home/nobr/prism/.venv/lib/python3.12/site-packages/gemma/gm/text/_tokenizer.py", line 161, in <module>
    class Tokenizer:
  File "/home/nobr/prism/.venv/lib/python3.12/site-packages/gemma/gm/text/_tokenizer.py", line 196, in Tokenizer
    FORMAT: ClassVar[dialog.Format] = dialog.Format.GEMMA3
                                      ^^^^^^^^^^^^^
AttributeError: module 'dialog' has no attribute 'Format'

[syrkis]

I'm a little confused about the best way to use Gemma in JAX? Is it tunix, maxtext or this repo?

[syrkis]

I don't mean to be a pest, but this is quite frustrating

[rastc]

pip install -U "git+https://github.com/google-deepmind/dialog.git@main"

you just need latest dialog