diff options
author | OTABI Tomoya <tomoya.otabi@gmail.com> | 2023-12-31 10:23:37 +0900 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-12-31 10:23:37 +0900 |
commit | 55b9aa64c39d6c34b6ef878fbad405900782e998 (patch) | |
tree | 411e73b35e04d7d2131d51f6ffee9c552749d7cd | |
parent | 1f98210a0f28f101728180aad5c27aa3adec54a6 (diff) | |
parent | 12a8aa5ac8b4d73ba03a171b6beaf2503a00bdcf (diff) |
Merge pull request #277788 from trofi/python3Packages.openllm-core-fix-optional-dependencies-eval
python3Packages.openllm-core: fix `optional-dependencies` evaluation
-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 b18017e10d183..ccf6aceef8bf3 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 |