Skip to content

Commit

Permalink
Update numba
Browse files Browse the repository at this point in the history
  • Loading branch information
@mawong-amd
mawong-amd committed Apr 10, 2024
1 parent 93ff699 commit 4357d8c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile.rocm
View file
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,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 \
Expand Down

0 comments on commit 4357d8c

Please sign in to comment.