search

google / gemma_pytorch

The official PyTorch implementation of Google's Gemma models
https://ai.google.dev/gemma
Apache License 2.0
5.19k stars 492 forks source link

Fix downcasting and upcasting similar to https://github.com/google/ge… #68

Closed michaelmoynihan closed 3 weeks ago

michaelmoynihan commented 3 weeks ago

Fix downcasting and upcasting similar to https://github.com/google/gemma_pytorch/pull/67, but for xla." [fix-xla-downcast ce4912d] Fix downcasting and upcasting similar to https://github.com/google/gemma_pytorch/pull/67, but for xla.

pengchongjin commented 3 weeks ago

Thanks @michaelmoynihan !