Skip to content

Support for Gemma-2 models #17

Open
@nerdimite

Description

@nerdimite

Hi, I wanted to ask if the support for Gemma2 models can be added in this binding? Specifically I was trying to load the gemma2-2b-it model but wasn't able to because the ModelType enum only supports 2b and 7b and passing gemma 2 weights in the path doesnt work.

Also, if it's possible, I think it would be better to make these bindings compatible directly with gemma.cpp's CLI args instead of custom enums mapping to an integer, so that this package in itself remains compatible with any new model types that come in the original gemma.cpp. Like directly passing the cpp args ./gemma --tokenizer tokenizer.spm --model gemma2-2b-it --compressed_weights 2.0-2b-it-sfp.sbs in the Gemma() class object

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions