mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2026-07-04 04:58:12 +08:00
28 lines
666 B
Python
28 lines
666 B
Python
from . import MLX, PyTorch
|
|
from .logger import console, logger, tb
|
|
from .MLX import quantization_methods_mlx
|
|
from .PyTorch import T2SEngineTorch, T2SRequest, T2SResult, quantization_methods_torch
|
|
from .PyTorch.structs import T2SEngineProtocol
|
|
|
|
backends = PyTorch.backends + MLX.backends
|
|
|
|
backends = [
|
|
b.replace("_", "-").title().replace("Mlx", "MLX").replace("Mps", "MPS").replace("Cuda", "CUDA") for b in backends
|
|
]
|
|
|
|
|
|
__all__ = [
|
|
"T2SEngineTorch",
|
|
"T2SRequest",
|
|
"T2SResult",
|
|
"backends",
|
|
"MLX",
|
|
"PyTorch",
|
|
"logger",
|
|
"console",
|
|
"tb",
|
|
"T2SEngineProtocol",
|
|
"quantization_methods_torch",
|
|
"quantization_methods_mlx",
|
|
]
|