Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.2 kB · 73 lines
Nix
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374{ lib, accelerate, buildPythonPackage, clean-fid, clip-anytorch, dctorch, einops, fetchFromGitHub, jsonmerge, kornia, pillow, pythonOlder, 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";
disabled = pythonOlder "3.7";
src = fetchFromGitHub { owner = "crowsonkb"; repo = "k-diffusion"; rev = "refs/tags/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 = with lib; { description = "Karras et al. (2022) diffusion models for PyTorch"; homepage = "https://github.com/crowsonkb/k-diffusion"; license = licenses.mit; maintainers = teams.tts.members; };}