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

   1 { lib
   2 , buildPythonPackage
   3 , fetchpatch
   4 , fetchFromGitHub
   5 , pythonAtLeast
   6
   7 , coqpit
   8 , fsspec
   9 , torch-bin
  10 , tensorboardx
  11 , protobuf
  12
  13 , pytestCheckHook
  14 , soundfile
  15 , torchvision-bin
  16 }:
  17
  18 let
  19   pname = "coqui-trainer";
  20   version = "0.0.16";
  21 in
  22 buildPythonPackage {
  23   inherit pname version;
  24   format = "pyproject";
  25
  26   src = fetchFromGitHub {
  27     owner = "coqui-ai";
  28     repo = "Trainer";
  29     rev = "refs/tags/v${version}";
  30     hash = "sha256-G2qFQEO5OOEI2jsb+zXsO8cbC13o7pqAC2xkFrhGkLc=";
  31   };
  32
  33   postPatch = ''
  34     sed -i 's/^protobuf.*/protobuf/' requirements.txt
  35   '';
  36
  37   propagatedBuildInputs = [
  38     coqpit
  39     fsspec
  40     torch-bin
  41     soundfile
  42     tensorboardx
  43     protobuf
  44   ];
  45
  46   # only one test and that requires training data from the internet
  47   doCheck = false;
  48
  49   checkInputs = [
  50     pytestCheckHook
  51     torchvision-bin
  52   ];
  53
  54   pythonImportsCheck = [
  55     "trainer"
  56   ];
  57
  58   meta = with lib; {
  59     description = "A general purpose model trainer, as flexible as it gets";
  60     homepage = "https://github.com/coqui-ai/Trainer";
  61     license = licenses.asl20;
  62     maintainers = teams.tts.members;
  63   };
  64 }