Something went wrong. Try again.
lol
Something went wrong. Try again.
1.1 kB · 69 lines
Nix
at master
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970{ lib, accelerate, buildPythonPackage, clean-fid, clip-anytorch, dctorch, einops, fetchFromGitHub, jsonmerge, kornia, pillow, rotary-embedding-torch, safetensors, scikit-image, scipy, torch, torchdiffeq, torchsde, torchvision, tqdm, wandb,}:
buildPythonPackage rec { pname = "k-diffusion"; version = "0.1.1.post1"; format = "setuptools";
src = fetchFromGitHub { owner = "crowsonkb"; repo = "k-diffusion"; tag = "v${version}"; hash = "sha256-x/UHzobQv5ov0luUHqC8OA5YbtF+aWL39/SQtzTm0RM="; };
propagatedBuildInputs = [ accelerate clean-fid clip-anytorch dctorch einops jsonmerge kornia pillow rotary-embedding-torch scikit-image scipy safetensors torch torchdiffeq torchsde torchvision tqdm wandb ];
pythonImportsCheck = [ "k_diffusion" ];
# no tests doCheck = false;
meta = { description = "Karras et al. (2022) diffusion models for PyTorch"; homepage = "https://github.com/crowsonkb/k-diffusion"; license = lib.licenses.mit; teams = [ lib.teams.tts ]; };}