summaryrefslogtreecommitdiff
path: root/makima/src/server/state.rs
diff options
context:
space:
mode:
Diffstat (limited to 'makima/src/server/state.rs')
-rw-r--r--makima/src/server/state.rs16
1 files changed, 8 insertions, 8 deletions
diff --git a/makima/src/server/state.rs b/makima/src/server/state.rs
index f662e30..bd6864f 100644
--- a/makima/src/server/state.rs
+++ b/makima/src/server/state.rs
@@ -560,7 +560,7 @@ pub struct ModelConfig {
pub parakeet_model_dir: String,
pub parakeet_eou_dir: String,
pub sortformer_model_path: String,
- pub qwen3_tts_dir: String,
+ pub chatterbox_model_dir: String,
}
/// Lazily-loaded ML models.
@@ -619,12 +619,12 @@ impl AppState {
/// * `parakeet_model_dir` - Path to the Parakeet TDT model directory
/// * `parakeet_eou_dir` - Path to the Parakeet EOU model directory
/// * `sortformer_model_path` - Path to the Sortformer diarization model file
- /// * `qwen3_tts_dir` - Path to the Qwen3-TTS model directory
+ /// * `chatterbox_model_dir` - Path to the Chatterbox TTS model directory
pub fn new(
parakeet_model_dir: &str,
parakeet_eou_dir: &str,
sortformer_model_path: &str,
- qwen3_tts_dir: &str,
+ chatterbox_model_dir: &str,
) -> Self {
// Create broadcast channels with buffer for 256 messages
let (file_updates, _) = broadcast::channel(256);
@@ -668,7 +668,7 @@ impl AppState {
parakeet_model_dir: parakeet_model_dir.to_string(),
parakeet_eou_dir: parakeet_eou_dir.to_string(),
sortformer_model_path: sortformer_model_path.to_string(),
- qwen3_tts_dir: qwen3_tts_dir.to_string(),
+ chatterbox_model_dir: chatterbox_model_dir.to_string(),
}),
ml_models: OnceCell::new(),
db_pool: None,
@@ -691,17 +691,17 @@ impl AppState {
/// Get or initialize the TTS engine (lazy loading).
///
- /// The TTS engine is loaded on first Speak connection using the Qwen3 backend.
+ /// The TTS engine is loaded on first Speak connection using the Chatterbox backend.
/// Returns a reference to the engine, or an error if loading fails.
pub async fn get_tts_engine(&self) -> Result<&dyn TtsEngine, Box<dyn std::error::Error + Send + Sync>> {
- let tts_dir = self.model_config.as_ref().map(|c| c.qwen3_tts_dir.as_str());
+ let tts_dir = self.model_config.as_ref().map(|c| c.chatterbox_model_dir.as_str());
self.tts_engine.get_or_try_init(|| async {
tracing::info!(
model_dir = ?tts_dir,
- "Lazy-loading TTS engine (Qwen3) on first Speak connection..."
+ "Lazy-loading TTS engine (Chatterbox) on first Speak connection..."
);
let engine = crate::tts::TtsEngineFactory::create(
- crate::tts::TtsBackend::Qwen3,
+ crate::tts::TtsBackend::Chatterbox,
tts_dir,
).map_err(|e| -> Box<dyn std::error::Error + Send + Sync> {
Box::new(e)