diff options
author | Theodore Ni <3806110+tjni@users.noreply.github.com> | 2022-12-04 21:46:07 -0800 |
---|---|---|
committer | Theodore Ni <3806110+tjni@users.noreply.github.com> | 2022-12-04 21:49:16 -0800 |
commit | 7833b95d8e30660533fe1ef0ec62715f18fcc65a (patch) | |
tree | 2c4ffee0e18faf6a2e57d73390ebd4ba9a812b79 | |
parent | bc85ef815830014d9deabb0803d46a78c832f944 (diff) |
python310Packages.invisible-watermark: pytorch -> torch
-rw-r--r-- | pkgs/development/python-modules/invisible-watermark/default.nix | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/invisible-watermark/default.nix b/pkgs/development/python-modules/invisible-watermark/default.nix index 82357741bab50..ce77b9e3115c2 100644 --- a/pkgs/development/python-modules/invisible-watermark/default.nix +++ b/pkgs/development/python-modules/invisible-watermark/default.nix @@ -3,7 +3,7 @@ , pythonOlder , fetchFromGitHub , opencv4 -, pytorch +, torch , onnx , onnxruntime , pillow @@ -14,6 +14,7 @@ buildPythonPackage rec { pname = "invisible-watermark"; version = "0.1.5"; + format = "setuptools"; disabled = pythonOlder "3.6"; src = fetchFromGitHub { @@ -25,7 +26,7 @@ buildPythonPackage rec { propagatedBuildInputs = [ opencv4 - pytorch + torch onnx onnxruntime pillow |