feat: numpy 2+ support (#937)

BBC-Esq · fumiama · web-flow · commit 1092c1ffcaa8 · 2025-05-23T21:59:02.000+09:00
Co-authored-by: 源文雨 &lt;41315874+fumiama@users.noreply.github.com&gt;
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -259,9 +259,10 @@ def infer(
             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
@@ -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
@@ -20,7 +20,7 @@
     license="AGPLv3+",
     install_requires=[
         "numba",
-        "numpy<2.0.0",
+        "numpy<3.0.0",
         "pybase16384",
         "torch>=2.1.0",
         "torchaudio",
