feat: numpy 2+ support (#937)

Co-authored-by: 源文雨 <41315874+fumiama@users.noreply.github.com>
2026-08-28 17:45:02 +08:00 · 2025-05-23 08:59:02 -04:00
parent 46ad65f903
commit 1092c1ffca
3 changed files with 4 additions and 3 deletions
@@ -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
@@ -1,4 +1,4 @@
-numpy<2.0.0
+numpy<3.0.0
 numba
 torch>=2.1.0
 torchaudio
@@ -20,7 +20,7 @@ setup(
    license="AGPLv3+",
    install_requires=[
        "numba",
-        "numpy<2.0.0",
+        "numpy<3.0.0",
        "pybase16384",
        "torch>=2.1.0",
        "torchaudio",