[Bugfix][ROCm] Add numba to Dockerfile.rocm (vllm-project#3962)

neuralmagic · Apr 11, 2024 · b454575 · b454575
1 parent d128ec9
commit b454575
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
@@ -91,8 +91,13 @@ RUN if [ "$BUILD_TRITON" = "1" ]; then \
 
 COPY ./ /app/vllm
 
+<<<<<<< HEAD
 RUN python3 -m pip install --upgrade pip
 RUN python3 -m pip install xformers==0.0.25 --no-deps
+=======
+RUN python3 -m pip install --upgrade pip numba
+RUN python3 -m pip install xformers==0.0.23 --no-deps
+>>>>>>> bd3c144e ([Bugfix][ROCm] Add numba to Dockerfile.rocm (#3962))
 
 RUN cd /app \
     && cd vllm \