42 lines
912 B
Nix
42 lines
912 B
Nix
# WARNING: This file was automatically generated. You should avoid editing it.
|
|
# If you run pynixify again, the file will be either overwritten or
|
|
# deleted, and you will lose the changes you made to it.
|
|
|
|
{ accelerate, buildPythonPackage, clip-anytorch, einops, fetchPypi, jsonmerge
|
|
, kornia, lib, pillow, resize-right, scikit-image, scipy, torch, torchdiffeq
|
|
, torchvision, tqdm, wandb, clean-fid, torchsde }:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "k-diffusion";
|
|
version = "0.1.0";
|
|
|
|
src = fetchPypi {
|
|
inherit pname version;
|
|
sha256 = "sha256-nsPzKQCtuyvCr/biuOVKx0ZyWfV1qU2m1z4aeoRhSPw=";
|
|
};
|
|
|
|
propagatedBuildInputs = [
|
|
accelerate
|
|
clip-anytorch
|
|
einops
|
|
jsonmerge
|
|
kornia
|
|
pillow
|
|
resize-right
|
|
scikit-image
|
|
scipy
|
|
torch
|
|
torchdiffeq
|
|
torchvision
|
|
torchsde
|
|
tqdm
|
|
wandb
|
|
clean-fid
|
|
];
|
|
|
|
# TODO FIXME
|
|
doCheck = false;
|
|
|
|
meta = with lib; { };
|
|
}
|