Skip to content

GitLab

Unverified Commit bd3c144e authored by Woosuk Kwon's avatar Woosuk Kwon Committed by GitHub
Browse files

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

parent 0258b7a9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
Dockerfile.rocm
View file @ bd3c144e
...@@ -91,7 +91,7 @@ RUN if [ "$BUILD_TRITON" = "1" ]; then \ ...@@ -91,7 +91,7 @@ RUN if [ "$BUILD_TRITON" = "1" ]; then \
COPY ./ /app/vllm 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 python3 -m pip install xformers==0.0.23 --no-deps
RUN cd /app \ RUN cd /app \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment