Kokoro-FastAPI/api/src/services/tts_model.py

import torch

if torch.cuda.is_available():
    from .tts_gpu import TTSGPUModel as TTSModel
else:
    from .tts_cpu import TTSCPUModel as TTSModel

__all__ = ["TTSModel"]
WIP, Functional for CPU: Updated for ONNX runtime support, Dockerfile and TTS Service 2025-01-03 00:53:41 -07:00			`import torch`

WIP: CPU/GPU Functional, few straggling tests to fix and check. 2025-01-03 03:16:42 -07:00			`if torch.cuda.is_available():`
			`from .tts_gpu import TTSGPUModel as TTSModel`
			`else:`
			`from .tts_cpu import TTSCPUModel as TTSModel`
WIP, Functional for CPU: Updated for ONNX runtime support, Dockerfile and TTS Service 2025-01-03 00:53:41 -07:00
WIP: CPU/GPU Functional, few straggling tests to fix and check. 2025-01-03 03:16:42 -07:00			`__all__ = ["TTSModel"]`