BLD: fix flashinfer installation in dockerfile (#2844)

xorbitsai · Feb 12, 2025 · 841cd26 · 841cd26
1 parent 617454c
commit 841cd26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xinference/deploy/docker/Dockerfile b/xinference/deploy/docker/Dockerfile
@@ -31,7 +31,7 @@ RUN pip install --upgrade -i "$PIP_INDEX" pip && \
     pip install -i "$PIP_INDEX" --upgrade-strategy only-if-needed -r /opt/inference/xinference/deploy/docker/requirements.txt && \
     pip install -i "$PIP_INDEX" --no-deps sglang && \
     pip uninstall flashinfer -y && \
-    pip install flashinfer_python -i https://flashinfer.ai/whl/cu124/torch2.5/flashinfer/ && \
+    pip install flashinfer-python -i https://flashinfer.ai/whl/cu124/torch2.5 && \
     cd /opt/inference && \
     python3 setup.py build_web && \
     git restore . && \