From 4131bcf7692323e9226dcc063f6e69c7c527e852 Mon Sep 17 00:00:00 2001 From: Dan Saunders Date: Sun, 28 Sep 2025 20:04:44 -0400 Subject: [PATCH] fix? --- docker/Dockerfile-uv-base | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile-uv-base b/docker/Dockerfile-uv-base index ff72ec50b..98a677b20 100644 --- a/docker/Dockerfile-uv-base +++ b/docker/Dockerfile-uv-base @@ -30,7 +30,7 @@ RUN uv venv --no-project --relocatable axolotl-venv ENV PATH="/workspace/axolotl-venv/bin:${PATH}" -RUN uv pip install --python "$VENV_PYTHON" packaging setuptools wheel psutil \ +RUN uv pip install --python "$VENV_PYTHON" packaging setuptools wheel psutil protobuf \ && uv pip install --python "$VENV_PYTHON" torch==${PYTORCH_VERSION} \ && uv pip install --python "$VENV_PYTHON" --no-build-isolation "causal_conv1d @ git+https://github.com/Dao-AILab/causal-conv1d.git@main" \ && uv pip install --python "$VENV_PYTHON" "mamba_ssm @ git+https://github.com/state-spaces/mamba.git@main" \