Make running preflight optional in model scripts

AI-Hypercomputer / maxtext

A simple, performant and scalable Jax LLM!

Apache License 2.0

1.53k stars 293 forks source link

Make running preflight optional in model scripts #867

Closed raymondzouu closed 2 months ago

raymondzouu commented 2 months ago

Make running preflight.sh optional in model scripts. This is needed so that these scripts can be ran in nightly pathways tests since rto setup does not work with pathways.