From 9190740e7738bc47cdce04df25fec8d24fb083a0 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Wed, 10 Apr 2024 06:39:14 +0000 Subject: [PATCH] [Bugfix][ROCm] Add numba to Dockerfile.rocm --- Dockerfile.rocm | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.rocm b/Dockerfile.rocm index 10b8bf1e7fabd..b1c5fac9d78ef 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -91,7 +91,7 @@ RUN if [ "$BUILD_TRITON" = "1" ]; then \ COPY ./ /app/vllm -RUN python3 -m pip install --upgrade pip +RUN python3 -m pip install --upgrade pip numba RUN python3 -m pip install xformers==0.0.23 --no-deps RUN cd /app \