tjh.dev
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
25buildPythonPackage 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}