From d3c143b63fd6c31860632021ca18f7b57cf19e74 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Tue, 21 Apr 2026 10:47:20 +0000
Subject: [PATCH] pocket-tts: 1.1.1 -> 2.0.0

Diff: https://github.com/kyutai-labs/pocket-tts/compare/v1.1.1...v2.0.0

Changelog: https://github.com/kyutai-labs/pocket-tts/releases/tag/v2.0.0
---
 pkgs/development/python-modules/pocket-tts/default.nix | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/pkgs/development/python-modules/pocket-tts/default.nix b/pkgs/development/python-modules/pocket-tts/default.nix
index 12bdc0787598..47da5a38b838 100644
--- a/pkgs/development/python-modules/pocket-tts/default.nix
+++ b/pkgs/development/python-modules/pocket-tts/default.nix
@@ -25,18 +25,20 @@
 
   # optional-dependencies
   soundfile,
+  torchao,
 }:
 
 buildPythonPackage (finalAttrs: {
   pname = "pocket-tts";
-  version = "1.1.1";
+  version = "2.0.0";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "kyutai-labs";
     repo = "pocket-tts";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-9Y/q/6Ti/oTyOgRzK+UT+LSM/iH3RGey+XojxbPvPS0=";
+    hash = "sha256-NbqL75EOS262L1km89raKXhgQVVOfi5yFP5Y4APH3yw=";
   };
 
   build-system = [
@@ -69,6 +71,9 @@ buildPythonPackage (finalAttrs: {
     audio = [
       soundfile
     ];
+    quantize = [
+      torchao
+    ];
   };
 
   pythonImportsCheck = [ "pocket_tts" ];