zero0303's picture
Add files using upload-large-folder tool
23c283a verified
Raw History Blame Contribute Delete
559 Bytes
"""Maya 2 Native (maya-research/maya2-native) on MLX: named and plain voices. See ../NOTES.md and ../README.md."""
import os
# The official engine runs its fp32 matmuls without TF32 (engine.py:155). MLX on the M5 runs fp32 matmuls and
# convolution GEMMs in TF32-class precision unless this is 0 (the earlier ports' measured lesson). Only fp32 work is
# affected here (the Mimi codec and the strict gates); a value already set by the caller is kept.
os.environ.setdefault("MLX_ENABLE_TF32", "0")
from .engine import Maya2 # noqa: E402
__all__ = ["Maya2"]