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

python3Packages.openllm: fix `optional-dependencies` attribute
This commit is contained in:
Mario Rodas 2024-01-13 16:07:43 -05:00 committed by GitHub
commit 24f24acb10
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 16 additions and 2 deletions

View File

@ -47,7 +47,7 @@ buildPythonPackage rec {
transformers
# diffusers
soundfile
];
] ++ transformers.optional-dependencies.agents;
full = passthru.optional-dependencies.grpc ++ passthru.optional-dependencies.agents;
};

View File

@ -143,7 +143,10 @@ buildPythonPackage rec {
# vllm
];
full = with passthru.optional-dependencies; (
agents ++ awq ++ baichuan ++ chatglm ++ ctranslate ++ falcon ++ fine-tune ++ ggml ++ gptq ++ mpt ++ openai ++ playground ++ starcoder ++ vllm
agents ++ awq ++ baichuan ++ chatglm ++ ctranslate ++ falcon ++ fine-tune ++ ggml ++ gptq ++ mpt
# disambiguate between derivation input and passthru field
++ passthru.optional-dependencies.openai
++ playground ++ starcoder ++ vllm
);
all = passthru.optional-dependencies.full;
};

View File

@ -15,9 +15,11 @@
, safetensors
, tqdm
# optional dependencies
, diffusers
, scikit-learn
, tensorflow
, onnxconverter-common
, opencv4
, tf2onnx
, torch
, accelerate
@ -87,6 +89,15 @@ buildPythonPackage rec {
vision = [ pillow ];
in
{
agents = [
diffusers
accelerate
datasets
torch
sentencepiece
opencv4
pillow
];
ja = [
# fugashi
# ipadic