diff --git a/docker/Dockerfile-uv-base b/docker/Dockerfile-uv-base
index 98a677b20..26bf6a772 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 protobuf \
+RUN uv pip install --python "$VENV_PYTHON" packaging setuptools wheel psutil protobuf grpclib \
     && 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" \