XXXXRT666 c1a4ff476c .
2025-10-19 21:51:54 +01:00

15 lines
457 B
Python

import importlib.util
import platform
if importlib.util.find_spec("mlx") is not None and platform.system() == "Darwin":
from .sample_funcs_mlx import sample_naive as sample_naive_mlx
from .t2s_engine_mlx import T2SEngine as T2SEngineMLX
backends = ["mlx_static", "mlx_varlen"]
else:
backends = []
quantization_methods_mlx = [None, "MXFP4", "Affine"]
__all__ = ["T2SEngineMLX", "sample_naive_mlx", "backends", "quantization_methods_mlx"]