diff options
-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 |