diff --git a/docker/Dockerfile.nmp-unsloth-training b/docker/Dockerfile.nmp-unsloth-training index 4abe4bc51d..19a68bf22e 100644 --- a/docker/Dockerfile.nmp-unsloth-training +++ b/docker/Dockerfile.nmp-unsloth-training @@ -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, @@ -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 @@ -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). @@ -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-* \ diff --git a/docker/Dockerfile.safe-synthesizer-tasks b/docker/Dockerfile.safe-synthesizer-tasks index 3330ae4088..76faf7a10f 100644 --- a/docker/Dockerfile.safe-synthesizer-tasks +++ b/docker/Dockerfile.safe-synthesizer-tasks @@ -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 diff --git a/docker/automodel/Dockerfile.nmp-automodel-base b/docker/automodel/Dockerfile.nmp-automodel-base index 39e83e6f69..fa395c403a 100644 --- a/docker/automodel/Dockerfile.nmp-automodel-base +++ b/docker/automodel/Dockerfile.nmp-automodel-base @@ -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" \ @@ -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-* \ diff --git a/plugins/nemo-safe-synthesizer/constraints.txt b/plugins/nemo-safe-synthesizer/constraints.txt index ae8172735b..cfa9473c7f 100644 --- a/plugins/nemo-safe-synthesizer/constraints.txt +++ b/plugins/nemo-safe-synthesizer/constraints.txt @@ -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