mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-09-29 17:10:02 +08:00
31 lines
585 B
Python
31 lines
585 B
Python
from . import MLX, PyTorch
|
|
from .logger import console, logger, tb
|
|
from .PyTorch import T2SEngineTorch, T2SRequest, T2SResult
|
|
from .PyTorch.structs import T2SEngineProtocol
|
|
|
|
backends = PyTorch.backends + MLX.backends
|
|
|
|
backends = [
|
|
b.replace("_", "-")
|
|
.title()
|
|
.replace("Mlx", "MLX")
|
|
.replace("Mps", "MPS")
|
|
.replace("Cuda", "CUDA")
|
|
.replace("Mxfp4", "MXFP4")
|
|
for b in backends
|
|
]
|
|
|
|
|
|
__all__ = [
|
|
"T2SEngineTorch",
|
|
"T2SRequest",
|
|
"T2SResult",
|
|
"backends",
|
|
"MLX",
|
|
"PyTorch",
|
|
"logger",
|
|
"console",
|
|
"tb",
|
|
"T2SEngineProtocol",
|
|
]
|