Python code for Gemma models

Hi, I would like to work on Gemma’s Python code and modify the architecture. Does anyone know where I can find the code? TIA

1 Like

This is not the code for the model itself, but I think this is the code for the class that handles the model.

1 Like