diff --git a/tts_service/src/inference/inference.rs b/tts_service/src/inference/inference.rs index 6c019917b1..40066fcbdb 100644 --- a/tts_service/src/inference/inference.rs +++ b/tts_service/src/inference/inference.rs @@ -8,7 +8,7 @@ pub trait InferencePipelineStart { type TtsModel; type VocoderModel; - fn clean_text(self, text: &str) + fn clean_text(self: Box, text: &str) -> AnyhowResult>>; } @@ -17,7 +17,7 @@ pub trait InferencePipelineTextCleaningDone { type TtsModel; type VocoderModel; - fn infer_mel(self, speaker_id: i64) + fn infer_mel(self: Box, speaker_id: i64) -> AnyhowResult>>; } diff --git a/tts_service/src/inference/pipelines/glowtts_melgan.rs b/tts_service/src/inference/pipelines/glowtts_melgan.rs index deb76c763a..1be05c1266 100644 --- a/tts_service/src/inference/pipelines/glowtts_melgan.rs +++ b/tts_service/src/inference/pipelines/glowtts_melgan.rs @@ -54,7 +54,7 @@ impl InferencePipelineStart for GlowTtsMelganPipeline { type TtsModel = Arc; type VocoderModel = Arc; - fn clean_text(self, text: &str) + fn clean_text(self: Box, text: &str) -> AnyhowResult>> { let cleaned_text = clean_text(text); @@ -71,7 +71,7 @@ impl InferencePipelineTextCleaningDone for GlowTtsMelganPipelineTextCleaningDone type TtsModel = Arc; type VocoderModel = Arc; - fn infer_mel(self, _speaker_id: i64) + fn infer_mel(self: Box, _speaker_id: i64) -> AnyhowResult>> { // TODO: Creating arpabet instances every time is inefficient (even if lazy_static! under the hood). diff --git a/tts_service/src/inference/pipelines/glowtts_multispeaker_melgan.rs b/tts_service/src/inference/pipelines/glowtts_multispeaker_melgan.rs index 54c8e6bb59..99c1117599 100644 --- a/tts_service/src/inference/pipelines/glowtts_multispeaker_melgan.rs +++ b/tts_service/src/inference/pipelines/glowtts_multispeaker_melgan.rs @@ -56,7 +56,7 @@ impl InferencePipelineStart for GlowTtsMultiSpeakerMelganPipeline { type TtsModel = Arc; type VocoderModel = Arc; - fn clean_text(self, text: &str) + fn clean_text(self: Box, text: &str) -> AnyhowResult>> { let cleaned_text = clean_text(text); @@ -73,7 +73,7 @@ impl InferencePipelineTextCleaningDone for GlowTtsMultiSpeakerMelganPipelineTex type TtsModel = Arc; type VocoderModel = Arc; - fn infer_mel(self, speaker_id: i64) + fn infer_mel(self: Box, speaker_id: i64) -> AnyhowResult>> { // TODO: Creating arpabet instances every time is inefficient (even if lazy_static! under the hood).