Luke Granger-Brown
57725ef3ec
git-subtree-dir: third_party/nixpkgs git-subtree-split: 76612b17c0ce71689921ca12d9ffdc9c23ce40b2
51 lines
966 B
Nix
51 lines
966 B
Nix
{ piper-tts
|
|
, python3
|
|
}:
|
|
|
|
let
|
|
python = python3;
|
|
in
|
|
|
|
python.pkgs.buildPythonPackage {
|
|
inherit (piper-tts) version src;
|
|
|
|
pname = "piper-train";
|
|
format = "setuptools";
|
|
|
|
sourceRoot = "${piper-tts.src.name}/src/python";
|
|
|
|
nativeBuildInputs = with python.pkgs; [
|
|
cython
|
|
];
|
|
|
|
postBuild = ''
|
|
make -C piper_train/vits/monotonic_align
|
|
'';
|
|
|
|
postInstall = ''
|
|
export MONOTONIC_ALIGN=$out/${python.sitePackages}/piper_train/vits/monotonic_align/monotonic_align
|
|
mkdir -p $MONOTONIC_ALIGN
|
|
cp -v ./piper_train/vits/monotonic_align/piper_train/vits/monotonic_align/core.*.so $MONOTONIC_ALIGN/
|
|
'';
|
|
|
|
propagatedBuildInputs = with python.pkgs; [
|
|
espeak-phonemizer
|
|
librosa
|
|
numpy
|
|
onnxruntime
|
|
piper-phonemize
|
|
pytorch-lightning
|
|
torch
|
|
];
|
|
|
|
pythonImportsCheck = [
|
|
"piper_train"
|
|
];
|
|
|
|
doCheck = false; # no tests
|
|
|
|
meta = piper-tts.meta // {
|
|
# requires torch<2, pytorch-lightning~=1.7
|
|
broken = true;
|
|
};
|
|
}
|