File size: 559 Bytes
23c283a | 1 2 3 4 5 6 7 8 9 10 11 12 | """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"]
|