wassname/vllm
1
0 Fork 0
mirror of https://github.com/wassname/vllm.git synced 2026-10-29 13:50:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Hardware][Intel CPU] Update torch 2.4.0 for CPU backend (#6931)

Browse Source
This commit is contained in:
Jie Fu (傅杰) authored and GitHub committed 2024-08-02 08:55:58 -07:00
1 parent db35186391
commit c16eaac500
2 files changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

Dockerfile.cpu
+1 -1
View File
@@ -26,7 +26,7 @@ COPY ./ /workspace/vllm
WORKDIR /workspace/vllm
RUN pip install -v -r requirements-cpu.txt --extra-index-url https://download.pytorch.org/whl/test/cpu
RUN pip install -v -r requirements-cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
# Support for building with non-AVX512 vLLM: docker build --build-arg VLLM_CPU_DISABLE_AVX512="true" ...
ARG VLLM_CPU_DISABLE_AVX512
requirements-cpu.txt
+1 -1
View File
@@ -2,5 +2,5 @@
-r requirements-common.txt
# Dependencies for x86_64 CPUs
torch == 2.4.0; platform_machine != "ppc64le"
torch == 2.4.0+cpu; platform_machine != "ppc64le"
torchvision; platform_machine != "ppc64le" # required for the image processor of phi3v, this must be updated alongside torch