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

is it possible to convert gemma_pytorch to onnx to tflite? #30

Open nyadla-sys opened 4 months ago

nyadla-sys commented 4 months ago

is it possible to convert gemma_pytorch to onnx to tflite?

suryabhupa commented 4 months ago

cc @pengchongjin any guidance here?

pengchongjin commented 4 months ago

Unfortunately, we haven't convert pytorch checkpoint to onnx or tflite ourselves.

Maybe you can follow the standard guides torch.onnx: https://pytorch.org/docs/stable/onnx.html and tf-lite to try the conversion.