Merge pull request #163321 from mweinelt/tts (4c7880fd) · Commits · nix / nixpkgs

pkgs/development/python-modules/coqui-trainer/default.nix

0 → 100644

+58 −0

Original line number	Diff line number	Diff line
		{ 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;
		};
		}

pkgs/tools/audio/tts/default.nix

+45 −26

Original line number	Diff line number	Diff line
		{ lib
		, python3
		, fetchFromGitHub
		, espeak-ng
		}:

		# USAGE:
		@@ -30,14 +31,14 @@ let
		in
		python.pkgs.buildPythonApplication rec {
		pname = "tts";
		version = "0.5.0";
		version = "0.6.1";
		format = "setuptools";

		src = fetchFromGitHub {
		owner = "coqui-ai";
		repo = "TTS";
		rev = "v${version}";
		sha256 = "sha256-9fNYNhHS9wqrk2bZnrkkGU1OaDu/16RA8fz+Zj9xsyQ=";
		sha256 = "sha256-YzMR/Tl1UvjdSqV/h4lYR6DuarEqEIM7RReqYznFU4Q=";
		};

		postPatch = let
		@@ -48,7 +49,7 @@ python.pkgs.buildPythonApplication rec {
		"numba"
		"numpy"
		"umap-learn"
		"torch"
		"unidic-lite"
		];
		in ''
		sed -r -i \
		@@ -65,6 +66,7 @@ python.pkgs.buildPythonApplication rec {
		propagatedBuildInputs = with python.pkgs; [
		anyascii
		coqpit
		coqui-trainer
		flask
		fsspec
		gdown
		@@ -92,29 +94,19 @@ python.pkgs.buildPythonApplication rec {
		];

		postInstall = ''
		cp -r TTS/server/templates/ $out/${python3.sitePackages}/TTS/server
		cp -r TTS/server/templates/ $out/${python.sitePackages}/TTS/server
		# cython modules are not installed for some reasons
		(
		cd TTS/tts/utils/monotonic_align
		${python3.interpreter} setup.py install --prefix=$out
		${python.interpreter} setup.py install --prefix=$out
		)
		'';

		checkInputs = with python3.pkgs; [
		pytest-sugar
		checkInputs = with python.pkgs; [
		espeak-ng
		pytestCheckHook
		];

		disabledTests = [
		# RuntimeError: fft: ATen not compiled with MKL support
		"test_torch_stft"
		"test_stft_loss"
		"test_multiscale_stft_loss"
		# Requires network acccess to download models
		"test_synthesize"
		"test_run_all_models"
		];

		preCheck = ''
		# use the installed TTS in $PYTHONPATH instead of the one from source to also have cython modules.
		mv TTS{,.old}
		@@ -125,22 +117,49 @@ python.pkgs.buildPythonApplication rec {

		for file in $(grep -rl 'python TTS/bin' tests); do
		substituteInPlace "$file" \
		--replace "python TTS/bin" "${python3.interpreter} $out/lib/${python3.libPrefix}/site-packages/TTS/bin"
		--replace "python TTS/bin" "${python.interpreter} $out/lib/${python.libPrefix}/site-packages/TTS/bin"
		done
		'';

		disabledTests = [
		# Requires network acccess to download models
		"test_synthesize"
		"test_run_all_models"
		# Mismatch between phonemes
		"test_text_to_ids_phonemes_with_eos_bos_and_blank"
		# Takes too long
		"test_parametrized_wavernn_dataset"
		];

		disabledTestPaths = [
		# requires tensorflow
		"tests/vocoder_tests/test_vocoder_tf_pqmf.py"
		"tests/vocoder_tests/test_vocoder_tf_melgan_generator.py"
		"tests/tts_tests/test_tacotron2_tf_model.py"
		# RuntimeError: fft: ATen not compiled with MKL support
		# phonemes mismatch between espeak-ng and gruuts phonemizer
		"tests/text_tests/test_phonemizer.py"
		# no training, it takes too long
		"tests/aux_tests/test_speaker_encoder_train.py"
		"tests/tts_tests/test_align_tts_train.py"
		"tests/tts_tests/test_fast_pitch_speaker_emb_train.py"
		"tests/tts_tests/test_fast_pitch_train.py"
		"tests/tts_tests/test_glow_tts_d-vectors_train.py"
		"tests/tts_tests/test_glow_tts_speaker_emb_train.py"
		"tests/tts_tests/test_glow_tts_train.py"
		"tests/tts_tests/test_speedy_speech_train.py"
		"tests/tts_tests/test_tacotron2_d-vectors_train.py"
		"tests/tts_tests/test_tacotron2_speaker_emb_train.py"
		"tests/tts_tests/test_tacotron2_train.py"
		"tests/tts_tests/test_tacotron2_train_fsspec_path.py"
		"tests/tts_tests/test_tacotron_train.py"
		"tests/tts_tests/test_vits_d-vectors_train.py"
		"tests/tts_tests/test_vits_multilingual_speaker_emb_train.py"
		"tests/tts_tests/test_vits_multilingual_train-d_vectors.py"
		"tests/tts_tests/test_vits_speaker_emb_train.py"
		"tests/tts_tests/test_vits_train.py"
		"tests/vocoder_tests/test_wavegrad_train.py"
		"tests/vocoder_tests/test_parallel_wavegan_train.py"
		"tests/vocoder_tests/test_fullband_melgan_train.py"
		"tests/vocoder_tests/test_hifigan_train.py"
		"tests/vocoder_tests/test_melgan_train.py"
		"tests/vocoder_tests/test_multiband_melgan_train.py"
		"tests/vocoder_tests/test_parallel_wavegan_train.py"
		"tests/vocoder_tests/test_melgan_train.py"
		"tests/vocoder_tests/test_wavernn_train.py"
		];

		meta = with lib; {
		@@ -148,6 +167,6 @@ python.pkgs.buildPythonApplication rec {
		changelog = "https://github.com/coqui-ai/TTS/releases/tag/v${version}";
		description = "Deep learning toolkit for Text-to-Speech, battle-tested in research and production";
		license = licenses.mpl20;
		maintainers = with maintainers; [ hexa mic92 ];
		maintainers = teams.tts.members;
		};
		}

pkgs/top-level/python-packages.nix

+2 −0

Original line number	Diff line number	Diff line
		@@ -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 { };

Admin message