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

Remove databricks-connect from all-cpu dep #1317

Closed XiaohanZhangCMU closed 4 days ago

XiaohanZhangCMU commented 5 days ago

Remove databricks-connect from deps to make it flexible use with applications running with databricks-connect, as the version mismatch can usually lead to strange import errors. Create a new extra deps which is the same as all-cpu except databricks-connect.