install_maxtext_extra_deps.install_post_train_extra_deps module

Contents

install_maxtext_extra_deps.install_post_train_extra_deps module#

Installs extra dependencies from a requirements file using uv.

This script is designed to be run to install dependencies specified in ‘extra_post_train_deps_from_github.txt’, which is expected to be in the same directory. It first ensures ‘uv’ is installed and then uses it to install the packages listed in the requirements file.

install_maxtext_extra_deps.install_post_train_extra_deps.main()[source]#

Installs extra dependencies specified in extra_post_train_deps_from_github.txt using uv.

This script looks for ‘extra_post_train_deps_from_github.txt’ relative to its own location. It executes ‘uv pip install -r <path_to_extra_deps.txt> –resolution=lowest’.