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

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