transformers

2a7e8e16 - [Examples] Add automatic dataset splitting in language-modeling examples (#9133)

[Examples] Add automatic dataset splitting in language-modeling examples (#9133) * replaced jnp.split + removing textual model inputs + ensuring warmup_steps > 0 * Add automatic dataset splitting in language-modeling examples