unsloth/docker
Daniel Han 8fa588db2c docker: pin xformers explicitly, forward the llama tag to the Studio build
The amd64 base install named the cu128-ampere-torch2110 extra, which does not
exist on main yet (the CUDA extras stop at torch2100): pip/uv only warn on an
unknown extra, so plain unsloth installed without xformers and the required-
package check failed the build. Both arches now take the plain huggingface
extra and amd64 pins xformers==0.0.35 explicitly in the same resolve (it
requires torch>=2.10 without an exact pin, pairing with the baked 2.11.0;
verified on PyPI, x86_64 wheels only, matching the arm64 exclusion). This
decouples the base image from the pending extras PR.

The Studio build now receives the SAME llama.cpp tag the base image baked:
Dockerfile.studio grows a LLAMA_PREBUILT_TAG arg exported as UNSLOTH_LLAMA_TAG
to install.sh (setup.sh honours it; the "latest" default is byte-identical to
setup.sh's own default for local builds), and the publish workflow forwards
the prepare job's resolved tag in the studio build-args. Without the pin a
dispatch override or an upstream release landing between the two jobs let the
no-GPU Studio build re-resolve "latest" and replace the pinned CUDA bundle.

The Studio venv-match assertion also needs installer support for torch 2.11
on the CUDA path; that lands in a separate installer PR and is now declared
as a merge-order dependency in the PR description (the publish workflow only
runs on main pushes, so nothing builds before both are merged).
2026-07-16 06:53:55 +00:00
..
jupyter docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
.dockerignore docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
build.sh docker: track latest llama.cpp release + show its update banner in Studio 2026-06-24 01:21:17 +00:00
Dockerfile docker: pin xformers explicitly, forward the llama tag to the Studio build 2026-07-16 06:53:55 +00:00
Dockerfile.studio docker: pin xformers explicitly, forward the llama tag to the Studio build 2026-07-16 06:53:55 +00:00
entrypoint.sh docker: preflight every GPU, classify all uv/pip value flags, volume-safe llama update 2026-07-16 05:51:15 +00:00
fetch_llama_prebuilt.py docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
NOTICE docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
run.sh docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
smoke_test.py docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
studio_launch.sh docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
supervisord.conf docker: optional Cloudflare tunnel for JupyterLab (UNSLOTH_JUPYTER_CLOUDFLARE) 2026-06-24 09:55:11 +00:00
unsloth_colab_compat.py docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
unsloth_ipython_startup.py docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
unsloth_jupyter_tunnel.sh docker: optional Cloudflare tunnel for JupyterLab (UNSLOTH_JUPYTER_CLOUDFLARE) 2026-06-24 09:55:11 +00:00
unsloth_llama_update.sh docker: preflight every GPU, classify all uv/pip value flags, volume-safe llama update 2026-07-16 05:51:15 +00:00
unsloth_nb_compat.py docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
unsloth_nb_content_sig.py docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
unsloth_nb_pip_magic.py docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
unsloth_nb_strip_colab.py docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
unsloth_nb_view.py docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
unsloth_pip_shim.py docker: strip VCS refs before the basename, bake the value-flag drift check into the build 2026-07-16 06:16:20 +00:00
unsloth_run.py docker: add the AGPL-3.0 SPDX header to the new Python files 2026-07-16 04:57:43 +00:00
unsloth_studio_update.sh docker: address review follow-ups (pip-shim flags, sync ownership, tags, zoo ref, arch list) 2026-06-26 09:06:39 +00:00
unsloth_sync_notebooks.sh docker: dedupe repeated rationale comments and parametrize the pip-shim tests 2026-07-16 05:27:22 +00:00