squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[FIX] Fix docker build error (#1831) (#1832)

Browse Source 
Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
This commit is contained in:
Allen 2023-11-30 15:06:50 +08:00 committed by GitHub
parent 63b2206ad0
commit f07c1ceaa5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Dockerfile
View File

@ -18,6 +18,11 @@ RUN --mount=type=cache,target=/root/.cache/pip \
# image to build pytorch extensions # image to build pytorch extensions
FROM dev AS build FROM dev AS build
# install build dependencies
COPY requirements-build.txt requirements-build.txt
RUN --mount=type=cache,target=/root/.cache/pip \
pip install -r requirements-build.txt
# copy input files # copy input files
COPY csrc csrc COPY csrc csrc
COPY setup.py setup.py COPY setup.py setup.py

3
pyproject.toml
View File

@ -1,8 +1,9 @@
[build-system] [build-system]
# Should be mirrored in requirements-build.txt
requires = [ requires = [
"ninja", "ninja",
"packaging", "packaging",
"setuptools", "setuptools >= 49.4.0",
"torch >= 2.1.0", "torch >= 2.1.0",
"wheel", "wheel",
] ]

6
requirements-build.txt Normal file
View File

@ -0,0 +1,6 @@
# Should be mirrored in pyproject.toml
ninja
packaging
setuptools>=49.4.0
torch>=2.1.0
wheel