···

       
17
       
17
       
 
       
  };

     

       
18
       
18
       
 
       
in

     

       
19
       
19
       
 
       
python.pkgs.buildPythonApplication rec {

     

       
20
       
20
       
-
       
  pname = "tts";

     

       
21
       
21
       
-
       
  version = "0.20.2";

     

       
20
       
20
       
+
       
  pname = "coqui-tts";

     

       
21
       
21
       
+
       
  version = "0.25.1";

     

       
22
       
22
       
 
       
  pyproject = true;

     

       
23
       
23
       
 
       


     

       
24
       
24
       
 
       
  src = fetchFromGitHub {

     

       
25
       
25
       
-
       
    owner = "coqui-ai";

     

       
26
       
26
       
-
       
    repo = "TTS";

     

       
25
       
25
       
+
       
    owner = "idiap";

     

       
26
       
26
       
+
       
    repo = "coqui-ai-TTS";

     

       
27
       
27
       
 
       
    rev = "refs/tags/v${version}";

     

       
28
       
28
       
-
       
    hash = "sha256-1nlSf15IEX1qKfDtR6+jQqskjxIuzaIWatkj9Z1fh8Y=";

     

       
28
       
28
       
+
       
    hash = "sha256-5w1Y9wdoJ+EV/WBwK3nqyY60NEsMjQsfE4g+sJB7VwQ=";

     

       
29
       
29
       
 
       
  };

     

       
30
       
30
       
 
       


     

       
31
       
31
       
 
       
  postPatch =

     
···

       
41
       
41
       
 
       
        "numpy"

     

       
42
       
42
       
 
       
        "unidic-lite"

     

       
43
       
43
       
 
       
        "trainer"

     

       
44
       
44
       
+
       
        "spacy\\[ja\\]"

     

       
45
       
45
       
+
       
        "transformers"

     

       
44
       
46
       
 
       
      ];

     

       
45
       
47
       
 
       
    in

     

       
46
       
48
       
 
       
    ''

     

       
47
       
49
       
 
       
      sed -r -i \

     

       
48
       
50
       
 
       
        ${lib.concatStringsSep "\n" (

     

       
49
       
49
       
-
       
          map (package: ''-e 's/${package}\s*[<>=]+.+/${package}/g' \'') relaxedConstraints

     

       
50
       
50
       
-
       
        )}

     

       
51
       
51
       
-
       
      requirements.txt

     

       
52
       
52
       
-
       


     

       
53
       
53
       
-
       
      sed -r -i \

     

       
54
       
54
       
-
       
        ${lib.concatStringsSep "\n" (

     

       
55
       
51
       
 
       
          map (package: ''-e 's/${package}\s*[<>=]+[^"]+/${package}/g' \'') relaxedConstraints

     

       
56
       
52
       
 
       
        )}

     

       
57
       
53
       
 
       
      pyproject.toml

     

       
58
       
58
       
-
       
      # only used for notebooks and visualization

     

       
59
       
59
       
-
       
      sed -r -i -e '/umap-learn/d' requirements.txt

     

       
60
       
54
       
 
       
    '';

     

       
61
       
55
       
 
       


     

       
62
       
56
       
 
       
  nativeBuildInputs = with python.pkgs; [

     
···

       
64
       
58
       
 
       
    numpy

     

       
65
       
59
       
 
       
    packaging

     

       
66
       
60
       
 
       
    setuptools

     

       
61
       
61
       
+
       
    hatchling

     

       
67
       
62
       
 
       
  ];

     

       
68
       
63
       
 
       


     

       
69
       
64
       
 
       
  propagatedBuildInputs = with python.pkgs; [

     
···

       
102
       
97
       
 
       
    transformers

     

       
103
       
98
       
 
       
    unidic-lite

     

       
104
       
99
       
 
       
    webrtcvad

     

       
100
       
100
       
+
       
    spacy

     

       
101
       
101
       
+
       
    monotonic-alignment-search

     

       
105
       
102
       
 
       
  ];

     

       
106
       
103
       
 
       


     

       
107
       
104
       
 
       
  postInstall = ''

     

       
108
       
105
       
 
       
    cp -r TTS/server/templates/ $out/${python.sitePackages}/TTS/server

     

       
109
       
109
       
-
       
    # cython modules are not installed for some reasons

     

       
110
       
110
       
-
       
    (

     

       
111
       
111
       
-
       
      cd TTS/tts/utils/monotonic_align

     

       
112
       
112
       
-
       
      ${python.pythonOnBuildForHost.interpreter} setup.py install --prefix=$out

     

       
113
       
113
       
-
       
    )

     

       
114
       
106
       
 
       
  '';

     

       
115
       
107
       
 
       


     

       
116
       
108
       
 
       
  # tests get stuck when run in nixpkgs-review, tested in passthru

     
···

       
161
       
153
       
 
       
    "tests/text_tests/test_phonemizer.py"

     

       
162
       
154
       
 
       
    # no training, it takes too long

     

       
163
       
155
       
 
       
    "tests/aux_tests/test_speaker_encoder_train.py"

     

       
164
       
164
       
-
       
    "tests/tts_tests/test_align_tts_train.py"

     

       
165
       
165
       
-
       
    "tests/tts_tests/test_fast_pitch_speaker_emb_train.py"

     

       
166
       
166
       
-
       
    "tests/tts_tests/test_fast_pitch_train.py"

     

       
167
       
167
       
-
       
    "tests/tts_tests/test_fastspeech_2_speaker_emb_train.py"

     

       
168
       
168
       
-
       
    "tests/tts_tests/test_fastspeech_2_train.py"

     

       
169
       
169
       
-
       
    "tests/tts_tests/test_glow_tts_d-vectors_train.py"

     

       
170
       
170
       
-
       
    "tests/tts_tests/test_glow_tts_speaker_emb_train.py"

     

       
171
       
171
       
-
       
    "tests/tts_tests/test_glow_tts_train.py"

     

       
156
       
156
       
+
       
    "tests/tts_tests2/test_align_tts_train.py"

     

       
157
       
157
       
+
       
    "tests/tts_tests2/test_fast_pitch_speaker_emb_train.py"

     

       
158
       
158
       
+
       
    "tests/tts_tests2/test_fast_pitch_train.py"

     

       
159
       
159
       
+
       
    "tests/tts_tests2/test_fastspeech_2_speaker_emb_train.py"

     

       
160
       
160
       
+
       
    "tests/tts_tests2/test_fastspeech_2_train.py"

     

       
161
       
161
       
+
       
    "tests/tts_tests2/test_glow_tts_d-vectors_train.py"

     

       
162
       
162
       
+
       
    "tests/tts_tests2/test_glow_tts_speaker_emb_train.py"

     

       
163
       
163
       
+
       
    "tests/tts_tests2/test_glow_tts_train.py"

     

       
172
       
164
       
 
       
    "tests/tts_tests/test_neuralhmm_tts_train.py"

     

       
173
       
165
       
 
       
    "tests/tts_tests/test_overflow_train.py"

     

       
174
       
166
       
 
       
    "tests/tts_tests/test_speedy_speech_train.py"

     
···

       
197
       
189
       
 
       
  };

     

       
198
       
190
       
 
       


     

       
199
       
191
       
 
       
  meta = with lib; {

     

       
200
       
200
       
-
       
    homepage = "https://github.com/coqui-ai/TTS";

     

       
201
       
201
       
-
       
    changelog = "https://github.com/coqui-ai/TTS/releases/tag/v${version}";

     

       
192
       
192
       
+
       
    homepage = "https://github.com/idiap/coqui-ai-TTS";

     

       
193
       
193
       
+
       
    changelog = "https://github.com/idiap/coqui-ai-TTS/releases/tag/v${version}";

     

       
202
       
194
       
 
       
    description = "Deep learning toolkit for Text-to-Speech, battle-tested in research and production";

     

       
203
       
195
       
 
       
    license = licenses.mpl20;

     

       
204
       
196
       
 
       
    maintainers = teams.tts.members;

     

       
205
       
205
       
-
       
    broken = true; # added 2024-04-08

     

       
197
       
197
       
+
       
    broken = false;

     

       
206
       
198
       
 
       
  };

     

       
207
       
199
       
 
       
}