diff options
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/k-diffusion/default.nix')
-rw-r--r-- | nixpkgs/pkgs/development/python-modules/k-diffusion/default.nix | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/k-diffusion/default.nix b/nixpkgs/pkgs/development/python-modules/k-diffusion/default.nix index 0d1410a5ced0..55a3ce36f80c 100644 --- a/nixpkgs/pkgs/development/python-modules/k-diffusion/default.nix +++ b/nixpkgs/pkgs/development/python-modules/k-diffusion/default.nix @@ -1,15 +1,14 @@ { lib -, buildPythonPackage -, fetchFromGitHub - -# dependencies , accelerate +, buildPythonPackage , clean-fid , clip-anytorch , einops +, fetchFromGitHub , jsonmerge , kornia , pillow +, pythonOlder , resize-right , scikit-image , scipy @@ -19,23 +18,24 @@ , torchvision , tqdm , wandb - }: buildPythonPackage rec { pname = "k-diffusion"; - version = "0.0.14"; + version = "0.0.16"; format = "setuptools"; + disabled = pythonOlder "3.7"; + src = fetchFromGitHub { owner = "crowsonkb"; repo = "k-diffusion"; - rev = "v${version}"; - hash = "sha256-KKVgk+1hidDBVaRnXjoqwuSRydI10OPHK3YModAizZU="; + rev = "refs/tags/v${version}"; + hash = "sha256-tOWDFt0/hGZF5HENiHPb9a2pBlXdSvDvCNTsCMZljC4="; }; propagatedBuildInputs = [ - accelerate + accelerate clean-fid clip-anytorch einops |