import numpy Float64DType to avoid Hunyuan Avatar to crash

import numpy Float64DType to avoid Hunyuan Avatar to crash if your numpy version is < 2.0
This commit is contained in:
unecologeek 2025-08-23 01:27:19 +02:00 committed by GitHub
parent ca4e96a8ae
commit e7dc11e612
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -41,7 +41,8 @@ if hasattr(torch.serialization, "add_safe_globals"): # TODO: this was added in
from numpy.core.multiarray import scalar
from numpy import dtype
from numpy.dtypes import Float64DType
import numpy as np
Float64DType = np.float64
from _codecs import encode
torch.serialization.add_safe_globals([ModelCheckpoint, scalar, dtype, Float64DType, encode])