mirror of
https://github.com/SebastianWendel/nixpkgs.git
synced 2024-09-20 12:29:02 +02:00
commit
2109b8d932
|
@ -36,14 +36,14 @@
|
||||||
|
|
||||||
buildPythonPackage rec {
|
buildPythonPackage rec {
|
||||||
pname = "vllm";
|
pname = "vllm";
|
||||||
version = "0.3.1";
|
version = "0.3.2";
|
||||||
format = "pyproject";
|
format = "pyproject";
|
||||||
|
|
||||||
src = fetchFromGitHub {
|
src = fetchFromGitHub {
|
||||||
owner = "vllm-project";
|
owner = "vllm-project";
|
||||||
repo = pname;
|
repo = pname;
|
||||||
rev = "v${version}";
|
rev = "v${version}";
|
||||||
hash = "sha256-hfd4ScU0mkZ7z4+w08BUA1K9bPXSiFThfiO+Ll2MTtg=";
|
hash = "sha256-ZFwlR8Xnen7FFblwzPJm0k+3iEo2p27QhfRaDfzwbOM=";
|
||||||
};
|
};
|
||||||
|
|
||||||
# Otherwise it tries to enumerate host supported ROCM gfx archs, and that is not possible due to sandboxing.
|
# Otherwise it tries to enumerate host supported ROCM gfx archs, and that is not possible due to sandboxing.
|
||||||
|
@ -62,11 +62,11 @@ buildPythonPackage rec {
|
||||||
substituteInPlace requirements.txt \
|
substituteInPlace requirements.txt \
|
||||||
--replace "cupy-cuda12x == 12.1.0" "cupy == 12.3.0"
|
--replace "cupy-cuda12x == 12.1.0" "cupy == 12.3.0"
|
||||||
substituteInPlace requirements-build.txt \
|
substituteInPlace requirements-build.txt \
|
||||||
--replace "torch==2.1.2" "torch == 2.2.0"
|
--replace "torch==2.1.2" "torch == 2.2.1"
|
||||||
substituteInPlace pyproject.toml \
|
substituteInPlace pyproject.toml \
|
||||||
--replace "torch == 2.1.2" "torch == 2.2.0"
|
--replace "torch == 2.1.2" "torch == 2.2.1"
|
||||||
substituteInPlace requirements.txt \
|
substituteInPlace requirements.txt \
|
||||||
--replace "torch == 2.1.2" "torch == 2.2.0"
|
--replace "torch == 2.1.2" "torch == 2.2.1"
|
||||||
'' + lib.optionalString rocmSupport ''
|
'' + lib.optionalString rocmSupport ''
|
||||||
substituteInPlace setup.py \
|
substituteInPlace setup.py \
|
||||||
--replace "'hipcc', '--version'" "'${writeShellScript "hipcc-version-stub" "echo HIP version: 0.0"}'"
|
--replace "'hipcc', '--version'" "'${writeShellScript "hipcc-version-stub" "echo HIP version: 0.0"}'"
|
||||||
|
|
Loading…
Reference in a new issue