mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2026-07-11 01:31:12 +08:00
24 lines
506 B
Python
24 lines
506 B
Python
from . import MLX, PyTorch
|
|
from .logger import 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") for b in backends
|
|
]
|
|
|
|
|
|
__all__ = [
|
|
"T2SEngineTorch",
|
|
"T2SRequest",
|
|
"T2SResult",
|
|
"backends",
|
|
"MLX",
|
|
"PyTorch",
|
|
"logger",
|
|
"tb",
|
|
"T2SEngineProtocol",
|
|
]
|