|
|
|
@ -1,15 +1,16 @@ |
|
|
|
|
{ lib |
|
|
|
|
, buildPythonPackage |
|
|
|
|
, fetchpatch |
|
|
|
|
, fetchFromGitHub |
|
|
|
|
, pythonAtLeast |
|
|
|
|
|
|
|
|
|
, coqpit |
|
|
|
|
, fsspec |
|
|
|
|
, pytorch |
|
|
|
|
, pytorch-bin |
|
|
|
|
|
|
|
|
|
, pytestCheckHook |
|
|
|
|
, soundfile |
|
|
|
|
, tensorboardx |
|
|
|
|
, torchvision |
|
|
|
|
, torchvision-bin |
|
|
|
|
}: |
|
|
|
|
|
|
|
|
|
let |
|
|
|
@ -20,6 +21,8 @@ buildPythonPackage { |
|
|
|
|
inherit pname version; |
|
|
|
|
format = "pyproject"; |
|
|
|
|
|
|
|
|
|
disabled = pythonAtLeast "3.10"; # https://github.com/coqui-ai/Trainer/issues/22 |
|
|
|
|
|
|
|
|
|
src = fetchFromGitHub { |
|
|
|
|
owner = "coqui-ai"; |
|
|
|
|
repo = "Trainer"; |
|
|
|
@ -27,12 +30,18 @@ buildPythonPackage { |
|
|
|
|
hash = "sha256-NsgCh+N2qWmRkTOjXqisVCP5aInH2zcNz6lsnIfVLiY="; |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
patches = [ |
|
|
|
|
(fetchpatch { |
|
|
|
|
url = "https://github.com/coqui-ai/Trainer/commit/07b447abf3290c8f2e5e723687b8a480b7382265.patch"; |
|
|
|
|
sha256 = "0v1hl784d9rghkblcfwgzp0gg9d6r5r0yv2kapzdz2qymiajy7y2"; |
|
|
|
|
}) |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
|
propagatedBuildInputs = [ |
|
|
|
|
coqpit |
|
|
|
|
fsspec |
|
|
|
|
pytorch |
|
|
|
|
pytorch-bin |
|
|
|
|
soundfile |
|
|
|
|
tensorboardx |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
|
# only one test and that requires training data from the internet |
|
|
|
@ -40,7 +49,7 @@ buildPythonPackage { |
|
|
|
|
|
|
|
|
|
checkInputs = [ |
|
|
|
|
pytestCheckHook |
|
|
|
|
torchvision |
|
|
|
|
torchvision-bin |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
|
pythonImportsCheck = [ |
|
|
|
|