diff options
| author | Sergei Trofimovich <slyich@gmail.com> | 2023-12-30 20:40:20 +0000 |
|---|---|---|
| committer | Sergei Trofimovich <slyich@gmail.com> | 2023-12-30 20:40:20 +0000 |
| commit | 12a8aa5ac8b4d73ba03a171b6beaf2503a00bdcf (patch) | |
| tree | 1a8c15fe30728570e12e89f72b910f17aed86a41 /pkgs | |
| parent | 01962add7e97c051a6c2dd2a013d7e7c8ec2388a (diff) | |
| download | nixlib-12a8aa5ac8b4d73ba03a171b6beaf2503a00bdcf.tar nixlib-12a8aa5ac8b4d73ba03a171b6beaf2503a00bdcf.tar.gz nixlib-12a8aa5ac8b4d73ba03a171b6beaf2503a00bdcf.tar.bz2 nixlib-12a8aa5ac8b4d73ba03a171b6beaf2503a00bdcf.tar.lz nixlib-12a8aa5ac8b4d73ba03a171b6beaf2503a00bdcf.tar.xz nixlib-12a8aa5ac8b4d73ba03a171b6beaf2503a00bdcf.tar.zst nixlib-12a8aa5ac8b4d73ba03a171b6beaf2503a00bdcf.zip | |
python3Packages.openllm-core: fix `optional-dependencies` evaluation
Without the change eval fails as:
$ nix build --no-link -f. python3Packages.openllm-core.optional-dependencies
error:
79| passthru.optional-dependencies =
| ^
80| let
error: attribute 'accelerate' missing
From what I see `accelerate` was always missing in `transflrmers` package.
While at it fixed `full` evaluation by disambiguating between input
argument and `passthru` attribute.
Diffstat (limited to 'pkgs')
| -rw-r--r-- | pkgs/development/python-modules/openllm-core/default.nix | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/pkgs/development/python-modules/openllm-core/default.nix b/pkgs/development/python-modules/openllm-core/default.nix index b18017e10d18..ccf6aceef8bf 100644 --- a/pkgs/development/python-modules/openllm-core/default.nix +++ b/pkgs/development/python-modules/openllm-core/default.nix @@ -70,9 +70,12 @@ buildPythonPackage rec { transformers # trl ] ++ transformers.optional-dependencies.torch - ++ transformers.optional-dependencies.tokenizers - ++ transformers.optional-dependencies.accelerate; - full = with passthru.optional-dependencies; ( vllm ++ bentoml ++ fine-tune ); + ++ transformers.optional-dependencies.tokenizers; + full = with passthru.optional-dependencies; ( + vllm + # use absolute path to disambiguate with derivbation argument + ++ passthru.optional-dependencies.bentoml + ++ fine-tune ); }; # there is no tests |