From 1334281d50e8cf478ee7f571b5fe12da1cd13cfa Mon Sep 17 00:00:00 2001 From: Dan Saunders Date: Sat, 30 Aug 2025 00:12:09 -0400 Subject: [PATCH] docker fix --- cicd/multigpu.py | 2 +- cicd/single_gpu.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cicd/multigpu.py b/cicd/multigpu.py index 5bd8d3c04..450a307f3 100644 --- a/cicd/multigpu.py +++ b/cicd/multigpu.py @@ -23,7 +23,7 @@ df_args = { "AXOLOTL_EXTRAS": os.environ.get("AXOLOTL_EXTRAS", ""), "AXOLOTL_ARGS": os.environ.get("AXOLOTL_ARGS", ""), "PYTORCH_VERSION": os.environ.get("PYTORCH_VERSION", "2.6.0"), - "BASE_TAG": os.environ.get("BASE_TAG", "main-base-py3.11-cu126-2.6.0"), + "BASE_TAG": os.environ.get("BASE_TAG", "main-base-uv-py3.11-cu126-2.6.0"), "CUDA": os.environ.get("CUDA", "126"), "GITHUB_REF": os.environ.get("GITHUB_REF", "refs/heads/main"), "GITHUB_SHA": os.environ.get("GITHUB_SHA", ""), diff --git a/cicd/single_gpu.py b/cicd/single_gpu.py index 5a06a34f0..f0d68ec42 100644 --- a/cicd/single_gpu.py +++ b/cicd/single_gpu.py @@ -23,7 +23,7 @@ df_args = { "AXOLOTL_EXTRAS": os.environ.get("AXOLOTL_EXTRAS", ""), "AXOLOTL_ARGS": os.environ.get("AXOLOTL_ARGS", ""), "PYTORCH_VERSION": os.environ.get("PYTORCH_VERSION", "2.6.0"), - "BASE_TAG": os.environ.get("BASE_TAG", "main-base-py3.11-cu126-2.6.0"), + "BASE_TAG": os.environ.get("BASE_TAG", "main-base-uv-py3.11-cu126-2.6.0"), "CUDA": os.environ.get("CUDA", "126"), "GITHUB_REF": os.environ.get("GITHUB_REF", "refs/heads/main"), "GITHUB_SHA": os.environ.get("GITHUB_SHA", ""),