Merge pull request #277751 from trofi/python3Packages.openllm-fix-optional-dependencies-eval
python3Packages.openllm: fix `optional-dependencies` attribute
This commit is contained in:
commit
24f24acb10
@ -47,7 +47,7 @@ buildPythonPackage rec {
|
||||
transformers
|
||||
# diffusers
|
||||
soundfile
|
||||
];
|
||||
] ++ transformers.optional-dependencies.agents;
|
||||
full = passthru.optional-dependencies.grpc ++ passthru.optional-dependencies.agents;
|
||||
};
|
||||
|
||||
|
@ -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;
|
||||
};
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user