python3Packages.transformers: fix pytorch reference

2022-05-30 00:02:51 -07:00 · 2022-05-30 00:02:51 -07:00 · d02116e319
commit d02116e319
parent b32b506ad0
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