fix: torch.from_numpy crash with numpy 2.x in frozen binary (#361)

mirror of https://github.com/jamiepine/voicebox.git synced 2026-10-03 00:55:14 -07:00

torch is compiled against numpy 1.x. numpy 2.x changed the ABI version
returned by PyArray_GetNDArrayCVersion() (0x01000009 → 0x02000000), so
torch's is_numpy_available() always returns False and torch.from_numpy()
raises RuntimeError. This causes TTS generation to fail with:

  ValueError: Unable to create tensor, you should probably activate
  padding with 'padding=True'

Two fixes:

1. Pin numpy<2.0 in requirements.txt so new builds bundle a compatible
   numpy version. (The existing comment already flagged this intention
   but the upper bound was never added.)

2. Add a PyInstaller runtime hook (pyi_rth_numpy_compat.py) that installs
   a ctypes memmove fallback for torch.from_numpy() at startup. Runtime
   hooks run after FrozenImporter is registered so frozen torch is
   importable. The fallback catches RuntimeError from the C-level ABI
   check and copies the numpy array into a new tensor via raw memory copy,
   bypassing the check entirely. This is a belt-and-suspenders fix that
   works regardless of the bundled numpy version.

Co-authored-by: aimaaaimaa <[email protected]>
Co-authored-by: Claude Sonnet 4.6 <[email protected]>

This commit is contained in:

aimaaaimaa

2026-04-16 01:46:40 -07:00

committed by

GitHub

co-authored by

aimaaaimaa

Claude Sonnet 4.6

parent 75abbb02c3

commit a383ff6863

3 changed files with 91 additions and 1 deletions

backend/requirements.txt

+1 -1

View File

@@ -50,7 +50,7 @@ en_core_web_sm @ https://github.com/explosion/spacy-models/releases/download/en_
 # Audio processing
 librosa>=0.10.0
 soundfile>=0.12.0
 numpy>=1.24.0
 numpy>=1.24.0,<2.0
 numba>=0.60.0,<0.61.0
 pedalboard>=0.9.0