diff --git a/pkgs/development/python-modules/openllm-client/default.nix b/pkgs/development/python-modules/openllm-client/default.nix index 082111392778..a1cc4fcc921e 100644 --- a/pkgs/development/python-modules/openllm-client/default.nix +++ b/pkgs/development/python-modules/openllm-client/default.nix @@ -47,7 +47,7 @@ buildPythonPackage rec { transformers # diffusers soundfile - ]; + ] ++ transformers.optional-dependencies.agents; full = passthru.optional-dependencies.grpc ++ passthru.optional-dependencies.agents; }; diff --git a/pkgs/development/python-modules/openllm/default.nix b/pkgs/development/python-modules/openllm/default.nix index b1cad320a11d..d889860e4e67 100644 --- a/pkgs/development/python-modules/openllm/default.nix +++ b/pkgs/development/python-modules/openllm/default.nix @@ -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; }; diff --git a/pkgs/development/python-modules/transformers/default.nix b/pkgs/development/python-modules/transformers/default.nix index c6d8e538e6a6..9168306bcb8a 100644 --- a/pkgs/development/python-modules/transformers/default.nix +++ b/pkgs/development/python-modules/transformers/default.nix @@ -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