summary refs log tree commit diff
path: root/pkgs/development/python-modules/coqui-trainer/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/coqui-trainer/default.nix')
-rw-r--r--pkgs/development/python-modules/coqui-trainer/default.nix15
1 files changed, 6 insertions, 9 deletions
diff --git a/pkgs/development/python-modules/coqui-trainer/default.nix b/pkgs/development/python-modules/coqui-trainer/default.nix
index a653370ad83..b7b08325e0b 100644
--- a/pkgs/development/python-modules/coqui-trainer/default.nix
+++ b/pkgs/development/python-modules/coqui-trainer/default.nix
@@ -7,6 +7,8 @@
 , coqpit
 , fsspec
 , pytorch-bin
+, tensorboardx
+, protobuf
 
 , pytestCheckHook
 , soundfile
@@ -15,7 +17,7 @@
 
 let
   pname = "coqui-trainer";
-  version = "0.0.11";
+  version = "0.0.12";
 in
 buildPythonPackage {
   inherit pname version;
@@ -25,21 +27,16 @@ buildPythonPackage {
     owner = "coqui-ai";
     repo = "Trainer";
     rev = "v${version}";
-    hash = "sha256-ujuQ9l6NOpDb2TdQbRcOM+j91RfbE8wCL9C0PID8g8Q=";
+    hash = "sha256-MSB3XbQALEKQi6Jtr/d2K8cIqyZryebYEcewGG48HV0=";
   };
 
-  patches = [
-    (fetchpatch {
-      url = "https://github.com/coqui-ai/Trainer/commit/07b447abf3290c8f2e5e723687b8a480b7382265.patch";
-      sha256 = "0v1hl784d9rghkblcfwgzp0gg9d6r5r0yv2kapzdz2qymiajy7y2";
-    })
-  ];
-
   propagatedBuildInputs = [
     coqpit
     fsspec
     pytorch-bin
     soundfile
+    tensorboardx
+    protobuf
   ];
 
   # only one test and that requires training data from the internet