From 84fee5ce2ae30fb2381c99b9b223b8235b962869 Mon Sep 17 00:00:00 2001 From: soryu Date: Sat, 20 Dec 2025 21:39:09 +0000 Subject: Remove TTS endpoint using chatterbox The library still remains, but this complicates deployment due to the large size of the model, so it is removed for now --- makima/src/server/state.rs | 7 ------- 1 file changed, 7 deletions(-) (limited to 'makima/src/server/state.rs') diff --git a/makima/src/server/state.rs b/makima/src/server/state.rs index 8eaf788..c38359d 100644 --- a/makima/src/server/state.rs +++ b/makima/src/server/state.rs @@ -4,7 +4,6 @@ use std::sync::Arc; use tokio::sync::Mutex; use crate::listen::{DiarizationConfig, ParakeetTDT, Sortformer}; -use crate::tts::ChatterboxTTS; /// Shared application state containing ML models. /// @@ -14,8 +13,6 @@ pub struct AppState { pub parakeet: Mutex, /// Speaker diarization model (Sortformer) pub sortformer: Mutex, - /// Text-to-speech model (ChatterboxTTS) - pub chatterbox: Mutex, } impl AppState { @@ -24,11 +21,9 @@ impl AppState { /// # Arguments /// * `parakeet_model_dir` - Path to the Parakeet STT model directory /// * `sortformer_model_path` - Path to the Sortformer diarization model file - /// * `tts_model_dir` - Optional path to the ChatterboxTTS model directory pub fn new( parakeet_model_dir: &str, sortformer_model_path: &str, - tts_model_dir: Option<&str>, ) -> Result> { let parakeet = ParakeetTDT::from_pretrained(parakeet_model_dir, None)?; let sortformer = Sortformer::with_config( @@ -36,12 +31,10 @@ impl AppState { None, DiarizationConfig::callhome(), )?; - let chatterbox = ChatterboxTTS::from_pretrained(tts_model_dir)?; Ok(Self { parakeet: Mutex::new(parakeet), sortformer: Mutex::new(sortformer), - chatterbox: Mutex::new(chatterbox), }) } } -- cgit v1.2.3