search

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.84k stars 503 forks source link

Add torch 2.3.1 docker images #1275

Closed dakinggg closed 3 weeks ago

dakinggg commented 3 weeks ago

Adds builds for 2.3.1 images. Future PRs will run CI on them and bump the min Foundry version. Also removes the "flash2" naming, since we only use flash2 at this point.