search

google / gemma_pytorch

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

Update run_xla.py #65

Closed michaelmoynihan closed 3 months ago

michaelmoynihan commented 3 months ago

Fixing run_xla.py. A tuple is returned from model and we need to ignore the logits in this script.