mosaicml / llm-foundry

LLM training code for Databricks foundation models
https://www.databricks.com/blog/introducing-dbrx-new-state-art-open-llm
Apache License 2.0
3.83k stars 502 forks source link

Bumping mlflow version to include buffering #1303

Closed JackZ-db closed 1 week ago

JackZ-db commented 1 week ago

The MlFlow version was previously capped at 2.14, leading to llm-foundry regression tests generating abnormal mlflow traffic. Updated to the correct version of MlFlow, at 'mlflow>=2.14.1,<3.0'.