pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 hatchling,
6
7 coqpit,
8 fsspec,
9 torch,
10 tensorboard,
11 protobuf,
12 psutil,
13
14 pytestCheckHook,
15 soundfile,
16 torchvision,
17}:
18
19let
20 pname = "coqui-tts-trainer";
21 version = "0.3.1";
22in
23buildPythonPackage {
24 inherit pname version;
25 format = "pyproject";
26
27 src = fetchFromGitHub {
28 owner = "idiap";
29 repo = "coqui-ai-Trainer";
30 tag = "v${version}";
31 hash = "sha256-vEVFnGn25F2lxG+oQzZWk20MarZdJrRkbsVC1rlEJwA=";
32 };
33
34 nativeBuildInputs = [
35 hatchling
36 ];
37
38 propagatedBuildInputs = [
39 coqpit
40 fsspec
41 protobuf
42 psutil
43 soundfile
44 tensorboard
45 torch
46 ];
47
48 # only one test and that requires training data from the internet
49 doCheck = false;
50
51 nativeCheckInputs = [
52 pytestCheckHook
53 torchvision
54 ];
55
56 pythonImportsCheck = [ "trainer" ];
57
58 meta = with lib; {
59 description = "General purpose model trainer, as flexible as it gets";
60 homepage = "https://github.com/idiap/coqui-ai-Trainer";
61 changelog = "https://github.com/idiap/coqui-ai-Trainer/releases/tag/v${version}";
62 license = licenses.asl20;
63 teams = [ teams.tts ];
64 };
65}