diff options
Diffstat (limited to 'pkgs/development/python-modules/transformers/default.nix')
-rw-r--r-- | pkgs/development/python-modules/transformers/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/transformers/default.nix b/pkgs/development/python-modules/transformers/default.nix index 97132a96455..864245622c7 100644 --- a/pkgs/development/python-modules/transformers/default.nix +++ b/pkgs/development/python-modules/transformers/default.nix @@ -17,7 +17,7 @@ , scikit-learn , pillow , pyyaml -, pytorch +, torch , tokenizers , tqdm }: @@ -67,7 +67,7 @@ buildPythonPackage rec { # tf2onnx ]; torch = [ - pytorch + torch ]; tokenizers = [ tokenizers |