AI-Hypercomputer / maxtext

A simple, performant and scalable Jax LLM!
Apache License 2.0
1.47k stars 275 forks source link

Adds a new end-to-end test for Mistral 7b #903

Closed shralex closed 1 day ago

shralex commented 5 days ago

The changes include:

The old test, which asserted on the generated text being equal, rather than logits, is removed.