Convert Gemma weights

[borisdayma]

Convert Gemma weights

[borisdayma]

A bit of progress:

# remove grain-nightly if installed from google/grain repo
pip uninstall grain-nightly

# install all requirements (seems a bit much just to convert weights…)
pip install -r requirements.txt

This works. Make sure to avoid conflicts with tensorflow versions (if you have a cpu version as well).

[rwitten]

I'm very confused by this thread Boris -- our tests all look green. I'm trying to imagine what the difference could be?

[borisdayma]

I initially ran the script, then saw I needed grain.
So then I tried pip install grain but it didn't work so I cloned google/grain` and installed from it which leads to this error.

The trick was to install grain-nightly. I'm closing it as it's resolved.