keras-nlp icon indicating copy to clipboard operation
keras-nlp copied to clipboard

Published 20 hours ago • verified publisher icon keras-team

Porting Gemma 2 transformers checkpoint

Open ariG23498 opened this issue 1 year ago • 1 comments

Porting Gemma 2 transformers checkpoints in Keras NLP

ariG23498 avatar Jun 27 '24 20:06 ariG23498

@mattdangerw @grasskin this PR is ready for review!

Note: The KerasNLP Gemma 2 model works only on the JAX backend (for the time being)

Also thanks to the Hugging Face team (Matt et. al.) for providing me with compute to work on this model.

ariG23498 avatar Jul 04 '24 07:07 ariG23498

Thanks!

mattdangerw avatar Jul 08 '24 17:07 mattdangerw