diff --git a/docker/Dockerfile b/docker/Dockerfile
index 2a7ede74a..9fe35da45 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -273,7 +273,6 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 RUN --mount=type=cache,target=/root/.cache/pip \
     python3 -m pip install \
     datamodel_code_generator \
-    mooncake-transfer-engine==0.3.8 \
     pre-commit \
     pytest \
     black \
@@ -286,6 +285,14 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     py-spy \
     cubloaty
 
+# Install mooncake
+RUN --mount=type=cache,target=/root/.cache/pip \
+    if [ "$GRACE_BLACKWELL" = "1" ]; then \
+        python3 -m pip install mooncake-transfer-engine==0.3.7.post2; \
+    else \
+        python3 -m pip install mooncake-transfer-engine==0.3.8; \
+    fi
+
 # Build and install sgl-model-gateway (install Rust, build, then remove to save space)
 RUN --mount=type=cache,target=/root/.cache/pip \
     curl --proto '=https' --tlsv1.2 --retry 3 --retry-delay 2 -sSf https://sh.rustup.rs | sh -s -- -y \