Merge pull request #277788 from trofi/python3Packages.openllm-core-fix-optional-dependencies-eval

python3Packages.openllm-core: fix `optional-dependencies` evaluation
This commit is contained in:
OTABI Tomoya 2023-12-31 10:23:37 +09:00 committed by GitHub
commit 55b9aa64c3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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