search

aws-neuron / transformers-neuronx

Apache License 2.0
100 stars 29 forks source link

Add support for `gemma` models #82

Open benglewis opened 7 months ago

benglewis commented 7 months ago

Hi 👋 ,

It would be really great if you could add support for the Gemma model series (i.e. 2B and 7B variants, particularly the 7B is what I would like most), since I see that it is currently not supported, but it is already supported by the mainstream transformers library.

cszhz commented 5 months ago

when will be gemma model supported? thanks.