Crate rwkv_tts_rs

Crate rwkv_tts_rs 

Source
Expand description

RWKV TTS Core Library

This library provides the core functionality for text-to-speech generation using RWKV models.

Re-exports§

pub use tts_state_manager::TtsInferContext;
pub use tts_state_manager::TtsInferOptions;
pub use tts_state_manager::TtsStateId;
pub use tts_state_manager::TtsStateManager;
pub use tts_state_manager::TtsStateStats;
pub use ref_audio_utilities::RefAudioUtilities;
pub use rwkv_sampler::RwkvSampler;
pub use rwkv_sampler::SamplerArgs;
pub use rwkv_sampler::TtsBatchRequest;
pub use properties_util::*;

Modules§

batch_types
批处理相关的共用类型定义和配置
dynamic_batch_manager
feature_extractor
特征提取模块
global_sampler_manager
inference_state_manager
推理状态管理器 - 优化推理调用频率
lightweight_tts_pipeline
轻量级TTS流水线 复用全局资源,不再每次创建新的模型实例
normal_mode_inference
onnx_session_pool
properties_util
属性处理工具模块 实现与Python版本properties_util.py相同的功能
ref_audio_utilities
参考音频处理工具模块 实现与Python版本ref_audio_utilities.py相同的功能
rwkv_sampler
RWKV模型推理采样器 实现基于web-rwkv库的RWKV模型推理和采样功能
sampler_manager
采样管理模块
shared_runtime
共享的RWKV运行时模块
streaming_inference
StreamingInference - 流式推理优化组件
tts_generator
TTS Generator module
tts_pipeline_fixes
TTS流水线修复模块 修复zero-shot模式下的问题,确保与C++实现完全一致
tts_state_manager
TTS状态管理器 参考ai00-core实现独立状态管理,每个请求获得独立状态副本
voice_feature
音色特征管理模块
voice_feature_manager
音色特征管理模块 实现音色特征的提取、保存、加载和管理功能
zero_shot_inference