Labyricorn/voicebox
1
0
Fork 0
mirror of https://github.com/jamiepine/voicebox.git synced 2026-10-02 16:45:15 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix(backend): numpy compat hook silently failed to patch torch

Browse Source 
dtype_map referenced _t, which is only bound as a default argument on
the inner function, so building the map raised NameError. The
surrounding except swallowed it and returned, meaning the from_numpy
fallback this hook exists for never applied in frozen builds.
This commit is contained in:
Jamie Pine
2026-07-26 23:15:53 -07:00
parent 3b0c29249b
commit 766c51a8a1
1 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/pyi_rth_numpy_compat.py
+12 -11
View File
@@ -44,6 +44,7 @@ def _patch_torch_from_numpy():
return return
try: try:
import ctypes import ctypes
import numpy as np import numpy as np
_orig = torch.from_numpy _orig = torch.from_numpy
@@ -53,17 +54,17 @@ def _patch_torch_from_numpy():
# silently corrupt data (e.g. fp16 tensors from some TTS engines), # silently corrupt data (e.g. fp16 tensors from some TTS engines),
# so we raise instead. # so we raise instead.
dtype_map = { dtype_map = {
"float16": _t.float16, "float16": torch.float16,
"float32": _t.float32, "float32": torch.float32,
"float64": _t.float64, "float64": torch.float64,
"int8": _t.int8, "int8": torch.int8,
"int16": _t.int16, "int16": torch.int16,
"int32": _t.int32, "int32": torch.int32,
"int64": _t.int64, "int64": torch.int64,
"uint8": _t.uint8, "uint8": torch.uint8,
"bool": _t.bool, "bool": torch.bool,
"complex64": _t.complex64, "complex64": torch.complex64,
"complex128": _t.complex128, "complex128": torch.complex128,
} }
def _safe_from_numpy( def _safe_from_numpy(