mirror of
https://github.com/remsky/Kokoro-FastAPI.git
synced 2025-04-13 09:39:17 +00:00
8 lines
172 B
Python
8 lines
172 B
Python
import torch
|
|
|
|
if torch.cuda.is_available():
|
|
from .tts_gpu import TTSGPUModel as TTSModel
|
|
else:
|
|
from .tts_cpu import TTSCPUModel as TTSModel
|
|
|
|
__all__ = ["TTSModel"]
|