diff options
| author | Martin Weinelt <hexa@darmstadt.ccc.de> | 2022-03-07 21:43:23 +0100 |
|---|---|---|
| committer | Martin Weinelt <hexa@darmstadt.ccc.de> | 2022-03-08 14:27:34 +0100 |
| commit | d96183955bfc6169d18fd8044198a5fd088de5ef (patch) | |
| tree | ad57b2fa328909c6c948eb5977ec995e4554645e | |
| parent | 1a0b8047999f117830a2c4b63362a807b49ca44e (diff) | |
| download | nixpkgs-d96183955bfc6169d18fd8044198a5fd088de5ef.tar nixpkgs-d96183955bfc6169d18fd8044198a5fd088de5ef.tar.gz nixpkgs-d96183955bfc6169d18fd8044198a5fd088de5ef.tar.bz2 nixpkgs-d96183955bfc6169d18fd8044198a5fd088de5ef.tar.lz nixpkgs-d96183955bfc6169d18fd8044198a5fd088de5ef.tar.xz nixpkgs-d96183955bfc6169d18fd8044198a5fd088de5ef.tar.zst nixpkgs-d96183955bfc6169d18fd8044198a5fd088de5ef.zip | |
python3Packages.coqui-trainer: init at 0.0.4
| -rw-r--r-- | pkgs/development/python-modules/coqui-trainer/default.nix | 58 | ||||
| -rw-r--r-- | pkgs/top-level/python-packages.nix | 2 |
2 files changed, 60 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/coqui-trainer/default.nix b/pkgs/development/python-modules/coqui-trainer/default.nix new file mode 100644 index 00000000000..5ff53b8eb75 --- /dev/null +++ b/pkgs/development/python-modules/coqui-trainer/default.nix @@ -0,0 +1,58 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub + +, coqpit +, fsspec +, pytorch-bin + +, pytestCheckHook +, soundfile +, tensorboardx +, torchvision +}: + +let + pname = "coqui-trainer"; + version = "0.0.4"; +in +buildPythonPackage { + inherit pname version; + format = "pyproject"; + + src = fetchFromGitHub { + owner = "coqui-ai"; + repo = "Trainer"; + # https://github.com/coqui-ai/Trainer/issues/4 + rev = "776eba829231543d3207927fc69b321d121e527c"; + hash = "sha256-ICveftJjBNsCgegTmd/ewd/Y6XGMg7YOvchx640RFPI="; + }; + + propagatedBuildInputs = [ + coqpit + fsspec + pytorch-bin + soundfile + tensorboardx + ]; + + # tests are failing; tests require the clearml library + # https://github.com/coqui-ai/Trainer/issues/5 + doCheck = false; + + checkInputs = [ + pytestCheckHook + torchvision + ]; + + pythonImportsCheck = [ + "trainer" + ]; + + meta = with lib; { + description = "A general purpose model trainer, as flexible as it gets"; + homepage = "https://github.com/coqui-ai/Trainer"; + license = licenses.asl20; + maintainers = teams.tts.members; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 7c08ff24e72..81696b5c9e7 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -1519,6 +1519,8 @@ in { coqpit = callPackage ../development/python-modules/coqpit { }; + coqui-trainer = callPackage ../development/python-modules/coqui-trainer {}; + cepa = callPackage ../development/python-modules/cepa { }; cerberus = callPackage ../development/python-modules/cerberus { }; |