From d02116e31944b63058b592b1025fe090eea6c95f Mon Sep 17 00:00:00 2001 From: Jonathan Ringer Date: Mon, 30 May 2022 00:02:51 -0700 Subject: [PATCH] python3Packages.transformers: fix pytorch reference --- pkgs/development/python-modules/transformers/default.nix | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkgs/development/python-modules/transformers/default.nix b/pkgs/development/python-modules/transformers/default.nix index adef7a19afcf..dcc89161d4ea 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 -, torch +, pytorch , tokenizers , tqdm }: @@ -67,7 +67,7 @@ buildPythonPackage rec { # tf2onnx ]; torch = [ - torch + pytorch ]; tokenizers = [ tokenizers