vs-overlay/plugins/vsgan/default.nix

43 lines
761 B
Nix
Raw Normal View History

{ 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;
};
}