huggingface / optimum-tpu

Google TPU optimizations for transformers models

Apache License 2.0

75 stars 19 forks source link

Closed tengomucho closed 1 month ago

tengomucho commented 1 month ago

What does this PR do?

Added support for gemma 7B and 2B with TGI and Jetstream Pytorch.

[x] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
[x] Did you make sure to update the documentation with your changes?
[x] Did you write any new necessary tests?