diff --git a/default.nix b/default.nix index 5801412..a616e58 100644 --- a/default.nix +++ b/default.nix @@ -67,6 +67,7 @@ in nnedi3_resample = callPythonPackage ./plugins/nnedi3_resample { }; nnedi3_rpow2 = callPythonPackage ./plugins/nnedi3_rpow2 { }; rekt = callPythonPackage ./plugins/rekt { }; + vsgan = callPythonPackage ./plugins/vsgan { }; vsTAAmbk = callPythonPackage ./plugins/vsTAAmbk { }; vsutil = callPythonPackage ./plugins/vsutil { }; diff --git a/plugins/vsgan/default.nix b/plugins/vsgan/default.nix new file mode 100644 index 0000000..deb0dcf --- /dev/null +++ b/plugins/vsgan/default.nix @@ -0,0 +1,42 @@ +{ lib, + buildPythonPackage, + fetchFromGitHub, + vapoursynth, + poetry, + python3Packages +}: + +buildPythonPackage rec { + pname = "vsgan"; + version = "1.6.4"; + + src = fetchFromGitHub { + owner = "rlaphoenix"; + repo = "VSGAN"; + rev = "v${version}"; + sha256 = "sha256-bMgCah3kkyxNU5tb/eLt0tuG4xnD4sbtAzUK0a4uOKE="; + }; + + format = "pyproject"; + + propagatedBuildInputs = [ + poetry + ] ++ (with python3Packages; [ + numpy + pytorch + ]); + + checkInputs = [ + vapoursynth + ]; + + pythonImportsCheck = [ "vsgan" ]; + + meta = with lib; { + description = "ESRGAN for VapourSynth"; + homepage = "https://vsgan.phoeniix.dev/"; + license = licenses.mit; + maintainers = with maintainers; [ aidalgol ]; + platforms = platforms.all; + }; +}