tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
python310Packages.k-diffusion: ini at 0.0.14
+71
+69
pkgs/development/python-modules/k-diffusion/default.nix
+69
pkgs/development/python-modules/k-diffusion/default.nix
···
1
+
{ lib
2
+
, buildPythonPackage
3
+
, fetchFromGitHub
4
+
5
+
# dependencies
6
+
, accelerate
7
+
, clean-fid
8
+
, clip-anytorch
9
+
, einops
10
+
, jsonmerge
11
+
, kornia
12
+
, pillow
13
+
, resize-right
14
+
, scikit-image
15
+
, scipy
16
+
, torch
17
+
, torchdiffeq
18
+
, torchsde
19
+
, torchvision
20
+
, tqdm
21
+
, wandb
22
+
23
+
}:
24
+
25
+
buildPythonPackage rec {
26
+
pname = "k-diffusion";
27
+
version = "0.0.14";
28
+
format = "setuptools";
29
+
30
+
src = fetchFromGitHub {
31
+
owner = "crowsonkb";
32
+
repo = "k-diffusion";
33
+
rev = "v${version}";
34
+
hash = "sha256-KKVgk+1hidDBVaRnXjoqwuSRydI10OPHK3YModAizZU=";
35
+
};
36
+
37
+
propagatedBuildInputs = [
38
+
accelerate
39
+
clean-fid
40
+
clip-anytorch
41
+
einops
42
+
jsonmerge
43
+
kornia
44
+
pillow
45
+
resize-right
46
+
scikit-image
47
+
scipy
48
+
torch
49
+
torchdiffeq
50
+
torchsde
51
+
torchvision
52
+
tqdm
53
+
wandb
54
+
];
55
+
56
+
pythonImportsCheck = [
57
+
"k_diffusion"
58
+
];
59
+
60
+
# no tests
61
+
doCheck = false;
62
+
63
+
meta = with lib; {
64
+
description = "Karras et al. (2022) diffusion models for PyTorch";
65
+
homepage = "https://github.com/crowsonkb/k-diffusion";
66
+
license = licenses.mit;
67
+
maintainers = teams.tts.members;
68
+
};
69
+
}
+2
pkgs/top-level/python-packages.nix
+2
pkgs/top-level/python-packages.nix
···
5324
5324
5325
5325
jxmlease = callPackage ../development/python-modules/jxmlease { };
5326
5326
5327
+
k-diffusion = callPackage ../development/python-modules/k-diffusion { };
5328
+
5327
5329
k5test = callPackage ../development/python-modules/k5test {
5328
5330
inherit (pkgs) krb5 findutils;
5329
5331
};