From 5771a65b88b1bef810826710965b29c7e7762264 Mon Sep 17 00:00:00 2001 From: Dan Saunders Date: Thu, 2 Oct 2025 11:20:23 -0400 Subject: [PATCH] fix --- cicd/Dockerfile.jinja | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cicd/Dockerfile.jinja b/cicd/Dockerfile.jinja index 9b8cba09a..46d0ca934 100644 --- a/cicd/Dockerfile.jinja +++ b/cicd/Dockerfile.jinja @@ -34,10 +34,10 @@ RUN if [ "$NIGHTLY_BUILD" = "true" ] ; then \ fi RUN uv pip install --system packaging==23.2 setuptools==75.8.0 -RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \ - uv sync --frozen --extra ring-flash-attn --extra optimizers --extra ray $(printf ' --extra %s' "${AXOLOTL_EXTRAS//,/ }") $AXOLOTL_ARGS; \ +RUN PYTHON_BIN="$(command -v python3)" && if [ "$AXOLOTL_EXTRAS" != "" ] ; then \ + uv sync --frozen --python "$PYTHON_BIN" --extra ring-flash-attn --extra optimizers --extra ray $(printf ' --extra %s' "${AXOLOTL_EXTRAS//,/ }") $AXOLOTL_ARGS; \ else \ - uv sync --frozen --extra ring-flash-attn --extra optimizers --extra ray $AXOLOTL_ARGS; \ + uv sync --frozen --python "$PYTHON_BIN" --extra ring-flash-attn --extra optimizers --extra ray $AXOLOTL_ARGS; \ fi RUN uv pip install --system --no-build-isolation flash-attn $AXOLOTL_ARGS