From 1092c1ffcaa82f4bdef104c35aa5541227c3e1d7 Mon Sep 17 00:00:00 2001
From: BBC-Esq <bbc@chintellalaw.com>
Date: Fri, 23 May 2025 08:59:02 -0400
Subject: [PATCH] feat: numpy 2+ support (#937)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: 源文雨 <41315874+fumiama@users.noreply.github.com>
---
 ChatTTS/core.py  | 3 ++-
 requirements.txt | 2 +-
 setup.py         | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/ChatTTS/core.py b/ChatTTS/core.py
index f1d7c47..0e79725 100644
--- a/ChatTTS/core.py
+++ b/ChatTTS/core.py
@@ -259,9 +259,10 @@ class Chat:
             return res_gen
         elif not refine_text_only:
             stripped_wavs = []
+            thr = np.float32(1e-5)
             for wavs in res_gen:
                 for wav in wavs:
-                    stripped_wavs.append(wav[np.abs(wav) > 1e-5])
+                    stripped_wavs.append(wav[np.abs(wav) > thr])
             if split_text:
                 return [np.concatenate(stripped_wavs)]
             return stripped_wavs
diff --git a/requirements.txt b/requirements.txt
index 75066bb..699194c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-numpy<2.0.0
+numpy<3.0.0
 numba
 torch>=2.1.0
 torchaudio
diff --git a/setup.py b/setup.py
index da7b609..dde50e3 100644
--- a/setup.py
+++ b/setup.py
@@ -20,7 +20,7 @@ setup(
     license="AGPLv3+",
     install_requires=[
         "numba",
-        "numpy<2.0.0",
+        "numpy<3.0.0",
         "pybase16384",
         "torch>=2.1.0",
         "torchaudio",