@@ -1,15 +1,16 @@
11{ lib
22, buildPythonPackage
3+, fetchpatch
34, fetchFromGitHub
5+, pythonAtLeast
46
57, coqpit
68, fsspec
7−, pytorch
9+, pytorch-bin
810
911, pytestCheckHook
1012, soundfile
11−, tensorboardx
12−, torchvision
13+, torchvision-bin
1314}:
1415
1516let
@@ -20,6 +21,8 @@ buildPythonPackage {
2021 inherit pname version;
2122 format = "pyproject";
2223
24+ disabled = pythonAtLeast "3.10"; # https://github.com/coqui-ai/Trainer/issues/22
25+
2326 src = fetchFromGitHub {
2427 owner = "coqui-ai";
2528 repo = "Trainer";
@@ -27,12 +30,18 @@ buildPythonPackage {
2730 hash = "sha256-NsgCh+N2qWmRkTOjXqisVCP5aInH2zcNz6lsnIfVLiY=";
2831 };
2932
33+ patches = [
34+ (fetchpatch {
35+ url = "https://github.com/coqui-ai/Trainer/commit/07b447abf3290c8f2e5e723687b8a480b7382265.patch";
36+ sha256 = "0v1hl784d9rghkblcfwgzp0gg9d6r5r0yv2kapzdz2qymiajy7y2";
37+ })
38+ ];
39+
3040 propagatedBuildInputs = [
3141 coqpit
3242 fsspec
33− pytorch
43+ pytorch-bin
3444 soundfile
35− tensorboardx
3645 ];
3746
3847 # only one test and that requires training data from the internet
@@ -40,7 +49,7 @@ buildPythonPackage {
4049
4150 checkInputs = [
4251 pytestCheckHook
43− torchvision
52+ torchvision-bin
4453 ];
4554
4655 pythonImportsCheck = [