Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.2 kB · 72 lines
Nix
at gcc-offload
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273{ 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; };}