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.99k stars 525 forks source link

Removing rich install #1198

Closed jjanezhang closed 4 months ago

jjanezhang commented 4 months ago

Removing rich install because rich is overriding the custom python excepthook we create in mcloud to catch exceptions.

This does mean that users submitting foundry runs through without mcli may not have rich formatting. However, if the run is submitted through mcli, our excepthook override will maintain rich formatting.

example of a run that maintains rich formatting after we remove the install and enable the mcloud excepthook

image