GPT-SoVITS/GPT_SoVITS/TTS_infer_pack/unified_engine_components.py
baicai-1145 a3a5aad157 Add unified engine components for TTS processing and state management
Introduce new modules including unified_engine_component_models, unified_engine_component_policy, unified_engine_component_registry, unified_engine_component_runtime, unified_engine_worker_completion, and unified_engine_worker_decode. These additions enhance the TTS framework by providing structured models for request handling, engine policies, and worker execution, significantly improving the architecture and maintainability of the system. The new components support asynchronous operations and optimize overall performance through better state management and processing capabilities.
2026-03-11 20:49:41 +08:00

64 lines
1.7 KiB
Python

from GPT_SoVITS.TTS_infer_pack.unified_engine_component_models import (
DirectTTSExecution,
NormalizedEngineRequest,
RuntimeControlCallbacks,
SchedulerDebugExecution,
SchedulerSubmitExecution,
)
from GPT_SoVITS.TTS_infer_pack.unified_engine_component_policy import (
EngineArbiterConfig,
EngineArbiterState,
EnginePolicyArbiterController,
EnginePolicyConfig,
)
from GPT_SoVITS.TTS_infer_pack.unified_engine_component_registry import (
DefaultReferenceState,
EngineRequestRegistry,
EngineRequestState,
EngineStatus,
ModelRegistry,
ModelRegistryState,
ReferenceRegistry,
SchedulerJobRegistry,
SchedulerPendingJob,
)
from GPT_SoVITS.TTS_infer_pack.unified_engine_component_runtime import (
EngineDecodeRuntimeOwner,
EngineDecodeRuntimeState,
EngineDispatchTask,
EngineFinalizeQueueState,
EngineGpuPrepareTask,
EngineTaskQueueOwner,
RuntimeStateCallbacks,
SchedulerFinalizeTask,
)
__all__ = [
"DefaultReferenceState",
"DirectTTSExecution",
"EngineArbiterConfig",
"EngineArbiterState",
"EngineDecodeRuntimeOwner",
"EngineDecodeRuntimeState",
"EngineDispatchTask",
"EngineFinalizeQueueState",
"EngineGpuPrepareTask",
"EnginePolicyArbiterController",
"EnginePolicyConfig",
"EngineRequestRegistry",
"EngineRequestState",
"EngineStatus",
"EngineTaskQueueOwner",
"ModelRegistry",
"ModelRegistryState",
"NormalizedEngineRequest",
"ReferenceRegistry",
"RuntimeControlCallbacks",
"RuntimeStateCallbacks",
"SchedulerDebugExecution",
"SchedulerFinalizeTask",
"SchedulerJobRegistry",
"SchedulerPendingJob",
"SchedulerSubmitExecution",
]