mosaicml / llm-foundry

LLM training code for Databricks foundation models

Home Page:https://www.databricks.com/blog/introducing-dbrx-new-state-art-open-llm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is there a way to figure out what dependencies are installed in the docker image?

sc-gr opened this issue · comments

E.g., if I'm using mosaicml/llm-foundry:2.2.1_cu121_flash2_aws-latest, is there a way to get the source of this docker image?

The image is built from the dockerfile (https://github.com/mosaicml/llm-foundry/blob/main/Dockerfile), with the base image coming from Composer (https://github.com/mosaicml/composer/tree/dev/docker). The simplest way to see everything in it would just be to make the container and then pip list to see the packages installed.