diff --git a/Dockerfile b/Dockerfile
index b6b429f..8d0805b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -238,4 +238,4 @@ RUN chmod +x $VLLM_BASE_DIR/run-cluster-node.sh
 
 # Final extra deps
 RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \
-    uv pip install ray[default] fastsafetensors
+    uv pip install ray[default] fastsafetensors "apache-tvm-ffi<0.2"
diff --git a/Dockerfile.wheels b/Dockerfile.wheels
index 1909a2f..3e42d47 100644
--- a/Dockerfile.wheels
+++ b/Dockerfile.wheels
@@ -94,5 +94,5 @@ RUN chmod +x $VLLM_BASE_DIR/run-cluster-node.sh
 
 # Final extra deps
 RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \
-    uv pip install ray[default] fastsafetensors
+    uv pip install ray[default] fastsafetensors "apache-tvm-ffi<0.2"