Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 10 additions & 4 deletions docker/Dockerfile.nmp-unsloth-training
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,8 @@
#
# Install steps:
# 1. `uv pip install unsloth --torch-backend=auto` plus explicit
# `transformers==5.3.0` and `huggingface-hub==1.3.0` pins. Unsloth's
# resolver allows transformers 5.3.0, and the explicit pins keep later
# `transformers==5.5.0` and `huggingface-hub==1.5.0` pins. Unsloth's
# resolver allows transformers 5.5.0, and the explicit pins keep later
# platform glue installs from re-solving the HF stack.
# Unsloth's resolver still pulls unsloth_zoo
# and the rest of the HF stack (trl, peft, accelerate, datasets,
Expand Down Expand Up @@ -71,8 +71,8 @@ ARG USERNAME=ubuntu
ARG USER_UID=1000
ARG USER_GID=1000
ARG UNSLOTH_VERSION=2026.6.1
ARG TRANSFORMERS_VERSION=5.3.0
ARG HF_HUB_VERSION=1.3.0
ARG TRANSFORMERS_VERSION=5.5.0
ARG HF_HUB_VERSION=1.5.0
ARG BITSANDBYTES_VERSION=0.49.2
ARG BNB_MAX_JOBS=10

Expand Down Expand Up @@ -177,6 +177,8 @@ RUN --mount=type=cache,target=/root/.cache/uv \
"aiohttp>=3.13.4,<4" \
"urllib3>=2.7.0,<3" \
"grpcio>=1.81.1,<2" \
"mistune>=3.3.3,<4" \
"soupsieve>=2.8.4,<3" \
"mlflow-skinny>=3.11.1,<3.12.0"

# Stale NGC system site-packages (verified on nvcr.io/nvidia/pytorch:26.05-py3).
Expand Down Expand Up @@ -205,6 +207,10 @@ RUN rm -rf \
/usr/local/lib/python3.12/dist-packages/mlflow \
/usr/local/lib/python3.12/dist-packages/mlflow-*.dist-info \
/usr/local/lib/python3.12/dist-packages/mlflow_skinny-*.dist-info \
/usr/local/lib/python3.12/dist-packages/mistune \
/usr/local/lib/python3.12/dist-packages/mistune-*.dist-info \
/usr/local/lib/python3.12/dist-packages/soupsieve \
/usr/local/lib/python3.12/dist-packages/soupsieve-*.dist-info \
/usr/local/lib/python3.12/dist-packages/jupyter_server \
/usr/local/lib/python3.12/dist-packages/jupyter_server-*.dist-info \
/usr/local/cuda/NsightSystems-cli-* \
Expand Down
2 changes: 1 addition & 1 deletion docker/Dockerfile.safe-synthesizer-tasks
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ RUN printf '%s\n' \
"${SAFE_SYNTHESIZER_RUNTIME_PACKAGE}" \
> /tmp/safe-synthesizer-runtime.txt && \
printf '%s\n' \
wandb==0.27.2 \
wandb==0.28.0 \
'cryptography>=48.0.1,<49' \
'pyarrow>=23.0.1,<24' \
> /tmp/safe-synthesizer-overrides.txt
Expand Down
11 changes: 8 additions & 3 deletions docker/automodel/Dockerfile.nmp-automodel-base
Original file line number Diff line number Diff line change
Expand Up @@ -97,13 +97,14 @@ RUN --mount=type=cache,target=/root/.cache/uv \
"GitPython>=3.1.50,<4" \
"jaraco-context>=6.1.0,<7" \
"jupyter-server>=2.20.0,<3" \
"jupyterlab>=4.5.7,<5" \
"mistune>=3.2.1,<4" \
"jupyterlab>=4.6.1,<5" \
"mistune>=3.3.3,<4" \
"notebook>=7.5.6,<8" \
"onnx>=1.21.0,<2" \
"pillow>=12.2.0,<13" \
"pillow>=12.3.0,<13" \
"pyasn1>=0.6.3,<1" \
"pyarrow>=23.0.1,<24" \
"soupsieve>=2.8.4,<3" \
"msgpack>=1.2.1,<2" \
"cryptography>=48.0.1,<49" \
"tornado>=6.5.5,<7" \
Expand Down Expand Up @@ -167,6 +168,10 @@ RUN rm -rf \
/usr/local/lib/python3.12/dist-packages/mlflow \
/usr/local/lib/python3.12/dist-packages/mlflow-*.dist-info \
/usr/local/lib/python3.12/dist-packages/mlflow_skinny-*.dist-info \
/usr/local/lib/python3.12/dist-packages/mistune \
/usr/local/lib/python3.12/dist-packages/mistune-*.dist-info \
/usr/local/lib/python3.12/dist-packages/soupsieve \
/usr/local/lib/python3.12/dist-packages/soupsieve-*.dist-info \
/usr/local/lib/python3.12/dist-packages/jupyter_server \
/usr/local/lib/python3.12/dist-packages/jupyter_server-*.dist-info \
/usr/local/cuda/NsightSystems-cli-* \
Expand Down
5 changes: 3 additions & 2 deletions plugins/nemo-safe-synthesizer/constraints.txt
Original file line number Diff line number Diff line change
Expand Up @@ -17,12 +17,13 @@ idna>=3.15
jsonpath-ng>=1.8.0
langchain-core>=1.2.28
langsmith>=0.7.31
mistune>=3.2.1
mistune>=3.3.3
nbconvert>=7.17.1
pandas<3
pillow>=12.2.0
pillow>=12.3.0
pyarrow>=23.0.1,<24
pymdown-extensions>=10.21.3
soupsieve>=2.8.4
python-dotenv>=1.2.2
python-multipart>=0.0.27
requests>=2.33.0
Expand Down