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 /pkgs/development/python-modules/coqui-trainer/default.nix | |
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
Diffstat (limited to 'pkgs/development/python-modules/coqui-trainer/default.nix')
-rw-r--r-- | pkgs/development/python-modules/coqui-trainer/default.nix | 58 |
1 files changed, 58 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; + }; +} |