sinapsis-speech 0.4.1__tar.gz → 0.4.3__tar.gz

{sinapsis_speech-0.4.1/packages/sinapsis_speech.egg-info → sinapsis_speech-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-speech
-Version: 0.4.1
+Version: 0.4.3
 Summary: Generate speech using various libraries.
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_speech-0.4.1 → sinapsis_speech-0.4.3}/packages/sinapsis_f5_tts/src/sinapsis_f5_tts/templates/f5_tts_inference.py

@@ -366,6 +366,6 @@ class F5TTSInference(Template):
         return container
 
     def reset_state(self, template_name: str | None = None) -> None:
-        if "cuda" in self.attributes.device:
+        if torch.cuda.is_available():
             torch.cuda.empty_cache()
         super().reset_state(template_name)

{sinapsis_speech-0.4.1 → sinapsis_speech-0.4.3/packages/sinapsis_speech.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-speech
-Version: 0.4.1
+Version: 0.4.3
 Summary: Generate speech using various libraries.
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_speech-0.4.1 → sinapsis_speech-0.4.3}/packages/sinapsis_zonos/src/sinapsis_zonos/templates/zonos_tts.py

@@ -109,16 +109,16 @@ class ZonosTTS(Template):
 
         Frees GPU memory by deleting the model and explicitly emptying the CUDA cache.
         """
-        if self.model:
-            del self.model
-            torch.cuda.empty_cache()
+
+        torch.cuda.empty_cache()
+        torch.cuda.ipc_collect()
 
     def reset_state(self, template_name: str | None = None) -> None:
         """Reinitialize the model and random seed."""
         _ = template_name
+        super().reset_state(template_name)
         self._del_model()
-        self.model = self._init_model()
-        init_seed(self.attributes)
+
         self.logger.debug(f"Model {self.attributes.model} reset\nSeed: {self.attributes.seed}")
 
     def generate_speech(self, input_data: list[TextPacket]) -> torch.Tensor:
@@ -157,7 +157,9 @@ class ZonosTTS(Template):
             container (DataContainer): The container to store metadata.
         """
         audio_np = output_audio[0].cpu().numpy()
-        container.audios.append(AudioPacket(content=audio_np, sample_rate=self.model.autoencoder.sampling_rate))
+        container.audios.append(
+            AudioPacket(content=audio_np.flatten(), sample_rate=self.model.autoencoder.sampling_rate)
+        )
 
     def execute(self, container: DataContainer) -> DataContainer:
         """Processes the input data and generates a speech output."""

{sinapsis_speech-0.4.1 → sinapsis_speech-0.4.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sinapsis-speech"
-version = "0.4.1"
+version = "0.4.3"
 description = "Generate speech using various libraries."
 authors = [
     {name = "SinapsisAI", email = "dev@sinapsis.tech"},