commit 67b3dc9bc9d2860df6e8fee90380903f476b37be · pyrox.dev/nixpkgs

+50

pkgs/development/python-modules/clean-fid/default.nix

··· 1 + { lib 2 + , buildPythonPackage 3 + , fetchFromGitHub 4 + 5 + # dependencies 6 + , numpy 7 + , pillow 8 + , requests 9 + , scipy 10 + , torch 11 + , torchvision 12 + , tqdm 13 + }: 14 + 15 + buildPythonPackage rec { 16 + pname = "clean-fid"; 17 + version = "0.1.35"; 18 + format = "setuptools"; 19 + 20 + src = fetchFromGitHub { 21 + owner = "GaParmar"; 22 + repo = "clean-fid"; 23 + rev = "c8ffa420a3923e8fd87c1e75170de2cf59d2644b"; 24 + hash = "sha256-fqBU/TmCXDTPU3KTP0+VYQoP+HsT2UMcZeLzQHKD9hw="; 25 + }; 26 + 27 + propagatedBuildInputs = [ 28 + numpy 29 + pillow 30 + requests 31 + scipy 32 + torch 33 + torchvision 34 + tqdm 35 + ]; 36 + 37 + pythonImportsCheck = [ 38 + "cleanfid" 39 + ]; 40 + 41 + # no tests1 42 + doCheck = false; 43 + 44 + meta = with lib; { 45 + description = "PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]"; 46 + homepage = "https://github.com/GaParmar/clean-fid"; 47 + license = licenses.mit; 48 + maintainers = teams.tts.members; 49 + }; 50 + }

+57

pkgs/development/python-modules/clip-anytorch/default.nix

··· 1 + { lib 2 + , buildPythonPackage 3 + , fetchFromGitHub 4 + 5 + # dependencies 6 + , ftfy 7 + , regex 8 + , tqdm 9 + , torch 10 + , torchvision 11 + 12 + # tests 13 + , pytestCheckHook 14 + }: 15 + 16 + buildPythonPackage rec { 17 + pname = "clip-anytorch"; 18 + version = "2.5.2"; 19 + format = "setuptools"; 20 + 21 + src = fetchFromGitHub { 22 + owner = "rom1504"; 23 + repo = "CLIP"; 24 + rev = version; 25 + hash = "sha256-EqVkpMQHawoCFHNupf49NrvLdGCq35wnYBpdP81Ztd4="; 26 + }; 27 + 28 + propagatedBuildInputs = [ 29 + ftfy 30 + regex 31 + tqdm 32 + torch 33 + torchvision 34 + ]; 35 + 36 + pythonImportsCheck = [ 37 + "clip" 38 + ]; 39 + 40 + # all tests require network access 41 + doCheck = false; 42 + 43 + nativeCheckInputs = [ 44 + pytestCheckHook 45 + ]; 46 + 47 + preCheck = '' 48 + export HOME=$TMPDIR 49 + ''; 50 + 51 + meta = with lib; { 52 + description = "Contrastive Language-Image Pretraining"; 53 + homepage = "https://github.com/rom1504/CLIP"; 54 + license = licenses.mit; 55 + maintainers = teams.tts.members; 56 + }; 57 + }

+69

pkgs/development/python-modules/k-diffusion/default.nix

··· 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 + 25 + buildPythonPackage 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 + }

+37

pkgs/development/python-modules/resize-right/default.nix

··· 1 + { lib 2 + , buildPythonPackage 3 + , fetchPypi 4 + 5 + # dependencies 6 + , numpy 7 + , torch 8 + }: 9 + 10 + buildPythonPackage rec { 11 + pname = "resize-right"; 12 + version = "0.0.2"; 13 + 14 + src = fetchPypi { 15 + inherit pname version; 16 + hash = "sha256-fcNbcs5AErd/fMkEmDUWN5OrmKWKuIk2EPsRn+Wa9SA="; 17 + }; 18 + 19 + propagatedBuildInputs = [ 20 + numpy 21 + torch 22 + ]; 23 + 24 + pythonImportsCheck = [ 25 + "resize_right" 26 + ]; 27 + 28 + # no tests 29 + doCheck = false; 30 + 31 + meta = with lib; { 32 + description = "The correct way to resize images or tensors. For Numpy or Pytorch (differentiable"; 33 + homepage = "https://github.com/assafshocher/ResizeRight"; 34 + license = licenses.mit; 35 + maintainers = teams.tts.members; 36 + }; 37 + }

+43

pkgs/development/python-modules/torchdiffeq/default.nix

··· 1 + { lib 2 + , buildPythonPackage 3 + , fetchPypi 4 + 5 + # dependencies 6 + , torch 7 + , scipy 8 + 9 + # tests 10 + , pytestCheckHook 11 + }: 12 + 13 + buildPythonPackage rec { 14 + pname = "torchdiffeq"; 15 + version = "0.2.3"; 16 + format = "setuptools"; 17 + 18 + src = fetchPypi { 19 + inherit pname version; 20 + hash = "sha256-/nX0NLkJCsDCdwLgK+0hRysPhwNb5lgfUe3F1AE+oxo="; 21 + }; 22 + 23 + propagatedBuildInputs = [ 24 + torch 25 + scipy 26 + ]; 27 + 28 + pythonImportsCheck = [ "torchdiffeq" ]; 29 + 30 + # no tests in sdist, no tags on git 31 + doCheck = false; 32 + 33 + nativeCheckInputs = [ 34 + pytestCheckHook 35 + ]; 36 + 37 + meta = with lib; { 38 + description = "Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation"; 39 + homepage = "https://github.com/rtqichen/torchdiffeq"; 40 + license = licenses.mit; 41 + maintainers = teams.tts.members; 42 + }; 43 + }

+67

pkgs/development/python-modules/torchsde/default.nix

··· 1 + { lib 2 + , buildPythonPackage 3 + , fetchFromGitHub 4 + 5 + # build-system 6 + , setuptools 7 + 8 + # dependencies 9 + , boltons 10 + , numpy 11 + , scipy 12 + , torch 13 + , trampoline 14 + 15 + # tests 16 + , pytestCheckHook 17 + }: 18 + 19 + buildPythonPackage rec { 20 + pname = "torchsde"; 21 + version = "0.2.4"; 22 + format = "pyproject"; 23 + 24 + src = fetchFromGitHub { 25 + owner = "google-research"; 26 + repo = "torchsde"; 27 + rev = "v${version}"; 28 + hash = "sha256-qQ7oswm0qTdq1xpQElt5cd3K0zskH+H/lgyEnxbCqsI="; 29 + }; 30 + 31 + postPatch = '' 32 + substituteInPlace setup.py \ 33 + --replace "numpy==1.19.*" "numpy" \ 34 + --replace "scipy==1.5.*" "scipy" 35 + ''; 36 + 37 + nativeBuildInputs = [ 38 + setuptools 39 + ]; 40 + 41 + propagatedBuildInputs = [ 42 + boltons 43 + numpy 44 + scipy 45 + torch 46 + trampoline 47 + ]; 48 + 49 + pythonImportsCheck = [ "torchsde" ]; 50 + 51 + nativeCheckInputs = [ 52 + pytestCheckHook 53 + ]; 54 + 55 + disabledTests = [ 56 + # RuntimeError: a view of a leaf Variable that requires grad is being used in an in-place operation. 57 + "test_adjoint" 58 + ]; 59 + 60 + meta = with lib; { 61 + changelog = "https://github.com/google-research/torchsde/releases/tag/v${version}"; 62 + description = "Differentiable SDE solvers with GPU support and efficient sensitivity analysis"; 63 + homepage = "https://github.com/google-research/torchsde"; 64 + license = licenses.asl20; 65 + maintainers = teams.tts.members; 66 + }; 67 + }

+34

pkgs/development/python-modules/trampoline/default.nix

··· 1 + { lib 2 + , buildPythonPackage 3 + , fetchFromGitLab 4 + , pytestCheckHook 5 + }: 6 + 7 + buildPythonPackage { 8 + pname = "trampoline"; 9 + version = "0.1.2"; 10 + format = "setuptools"; 11 + 12 + # only wheel on pypi, no tags on git 13 + src = fetchFromGitLab { 14 + owner = "ferreum"; 15 + repo = "trampoline"; 16 + rev = "1d98f39c3015594e2ac8ed48dccc2f393b4dd82b"; 17 + hash = "sha256-A/tuR+QW9sKh76Qjwn1uQxlVJgWrSFzXeBRDdnSi2o4="; 18 + }; 19 + 20 + pythonImportsCheck = [ 21 + "trampoline" 22 + ]; 23 + 24 + nativeCheckInputs = [ 25 + pytestCheckHook 26 + ]; 27 + 28 + meta = with lib; { 29 + description = "Simple and tiny yield-based trampoline implementation for python"; 30 + homepage = "https://gitlab.com/ferreum/trampoline"; 31 + license = licenses.mit; 32 + maintainers = teams.tts.members; 33 + }; 34 + }

+7 -2

pkgs/tools/audio/tts/default.nix

··· 8 8 let 9 9 python = python3.override { 10 10 packageOverrides = self: super: { 11 + torch = super.torch-bin; 12 + torchvision = super.torchvision-bin; 11 13 }; 12 14 }; 13 15 in 14 16 python.pkgs.buildPythonApplication rec { 15 17 pname = "tts"; 16 - version = "0.13.3"; 18 + version = "0.14.0"; 17 19 format = "pyproject"; 18 20 19 21 src = fetchFromGitHub { 20 22 owner = "coqui-ai"; 21 23 repo = "TTS"; 22 24 rev = "refs/tags/v${version}"; 23 - hash = "sha256-cu714/XtVqqlHN2CmUObcNFG6Vdi9VqC4at/HB8euDs="; 25 + hash = "sha256-AVU4ULz++t9850pYeNrG5HKNvUZcMld4O1/zu697rzk="; 24 26 }; 25 27 26 28 postPatch = let ··· 57 59 bnnumerizer 58 60 bnunicodenormalizer 59 61 coqpit 62 + einops 60 63 flask 61 64 fsspec 62 65 g2pkk ··· 65 68 inflect 66 69 jamo 67 70 jieba 71 + k-diffusion 68 72 librosa 69 73 matplotlib 70 74 mecab-python3 ··· 81 85 torchaudio-bin 82 86 tqdm 83 87 trainer 88 + transformers 84 89 unidic-lite 85 90 webrtcvad 86 91 ];

+14

pkgs/top-level/python-packages.nix

··· 1831 1831 1832 1832 cle = callPackage ../development/python-modules/cle { }; 1833 1833 1834 + clean-fid = callPackage ../development/python-modules/clean-fid { }; 1835 + 1834 1836 cleo = callPackage ../development/python-modules/cleo { }; 1835 1837 1836 1838 clevercsv = callPackage ../development/python-modules/clevercsv { }; ··· 1838 1840 clf = callPackage ../development/python-modules/clf { }; 1839 1841 1840 1842 clip = callPackage ../development/python-modules/clip { }; 1843 + 1844 + clip-anytorch = callPackage ../development/python-modules/clip-anytorch { }; 1841 1845 1842 1846 cock = callPackage ../development/python-modules/cock { }; 1843 1847 ··· 5319 5323 jwcrypto = callPackage ../development/python-modules/jwcrypto { }; 5320 5324 5321 5325 jxmlease = callPackage ../development/python-modules/jxmlease { }; 5326 + 5327 + k-diffusion = callPackage ../development/python-modules/k-diffusion { }; 5322 5328 5323 5329 k5test = callPackage ../development/python-modules/k5test { 5324 5330 inherit (pkgs) krb5 findutils; ··· 10463 10469 10464 10470 resampy = callPackage ../development/python-modules/resampy { }; 10465 10471 10472 + resize-right = callPackage ../development/python-modules/resize-right { }; 10473 + 10466 10474 resolvelib = callPackage ../development/python-modules/resolvelib { }; 10467 10475 10468 10476 responses = callPackage ../development/python-modules/responses { }; ··· 12147 12155 12148 12156 torchaudio-bin = callPackage ../development/python-modules/torchaudio/bin.nix { }; 12149 12157 12158 + torchdiffeq = callPackage ../development/python-modules/torchdiffeq { }; 12159 + 12150 12160 torchgpipe = callPackage ../development/python-modules/torchgpipe { }; 12151 12161 12152 12162 torchmetrics = callPackage ../development/python-modules/torchmetrics { }; ··· 12156 12166 torchinfo = callPackage ../development/python-modules/torchinfo { }; 12157 12167 12158 12168 torchlibrosa = callPackage ../development/python-modules/torchlibrosa { }; 12169 + 12170 + torchsde = callPackage ../development/python-modules/torchsde { }; 12159 12171 12160 12172 torchvision = callPackage ../development/python-modules/torchvision { }; 12161 12173 ··· 12202 12214 traitsui = callPackage ../development/python-modules/traitsui { }; 12203 12215 12204 12216 traittypes = callPackage ../development/python-modules/traittypes { }; 12217 + 12218 + trampoline = callPackage ../development/python-modules/trampoline { }; 12205 12219 12206 12220 transaction = callPackage ../development/python-modules/transaction { }; 12207 12221