Merge pull request #175438 from jonringer/fix-transformers-pytorch

python3Packages.transformers: fix pytorch reference
2022-05-30 09:47:08 +02:00 · 2022-05-30 09:47:08 +02:00 · 33c6d916c7
parent c68d933481 d02116e319
commit 33c6d916c7
1 changed files with 2 additions and 2 deletions
--- a/pkgs/development/python-modules/transformers/default.nix
+++ b/pkgs/development/python-modules/transformers/default.nix
@ -17,7 +17,7 @@
 , scikit-learn
 , pillow
 , pyyaml
-, torch
+, pytorch
 , tokenizers
 , tqdm
 }:
@ -67,7 +67,7 @@ buildPythonPackage rec {
      # tf2onnx
    ];
    torch = [
-      torch
+      pytorch
    ];
    tokenizers = [
      tokenizers