sinapsis-speech 0.4.1__py3-none-any.whl → 0.4.3__py3-none-any.whl

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.dist-info → sinapsis_speech-0.4.3.dist-info}/METADATA

@@ -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.dist-info → sinapsis_speech-0.4.3.dist-info}/RECORD

@@ -13,7 +13,7 @@ sinapsis_f5_tts/src/sinapsis_f5_tts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 sinapsis_f5_tts/src/sinapsis_f5_tts/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_f5_tts/src/sinapsis_f5_tts/helpers/tags.py,sha256=T9RbgsPgq7Wc-0Lu3W8Si4bxjQsyWbX6hPrc-dakdCs,208
 sinapsis_f5_tts/src/sinapsis_f5_tts/templates/__init__.py,sha256=28BOPAr9GG1jYcrXi45ZWO1n2FAZJOdDcmRkOXdEYmk,496
-sinapsis_f5_tts/src/sinapsis_f5_tts/templates/f5_tts_inference.py,sha256=D__uML5TyOj-KHN6S5RgkBynkYk250s-4EawuiNSh3I,16351
+sinapsis_f5_tts/src/sinapsis_f5_tts/templates/f5_tts_inference.py,sha256=HooFxYB1lqRKuzGjMDiw48Xzm34YI81dE0uD3WWm89A,16344
 sinapsis_kokoro/src/sinapsis_kokoro/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_kokoro/src/sinapsis_kokoro/helpers/kokoro_utils.py,sha256=2IMJuwURPKK7keIkgS-rpGD28REG5M1FwW0COGcm3nI,1573
 sinapsis_kokoro/src/sinapsis_kokoro/helpers/tags.py,sha256=AiHIvqmujKX6tYQ4lEXjRGhq8Ujst8gZwVmcAjS7u3k,210
@@ -30,15 +30,15 @@ sinapsis_parakeet_tdt/src/sinapsis_parakeet_tdt/helpers/__init__.py,sha256=47DEQ
 sinapsis_parakeet_tdt/src/sinapsis_parakeet_tdt/helpers/tags.py,sha256=OKZbq4zIL6XWM7eG5WuQ3dWYkmYNWjuCnlseXmjR_j0,262
 sinapsis_parakeet_tdt/src/sinapsis_parakeet_tdt/templates/__init__.py,sha256=3LppgbS6v70Rmx__yXXQgnoZ2ZBHcXkXeWZYQQf6Zwg,504
 sinapsis_parakeet_tdt/src/sinapsis_parakeet_tdt/templates/parakeet_tdt.py,sha256=ECNDd_zxkkop3WikQpkRW_bjLAMjROGpdWbChvaJdBE,9948
-sinapsis_speech-0.4.1.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+sinapsis_speech-0.4.3.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
 sinapsis_zonos/src/sinapsis_zonos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_zonos/src/sinapsis_zonos/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_zonos/src/sinapsis_zonos/helpers/tags.py,sha256=Y7MKQNx1E2k7ebF6r_1l1nBeS5k8hO424yFTT9NI7Rg,244
 sinapsis_zonos/src/sinapsis_zonos/helpers/zonos_keys.py,sha256=m1GdOYfzP73JGmtxH30mNiqbNkzFsQl9o2QaT7QxSVU,2470
 sinapsis_zonos/src/sinapsis_zonos/helpers/zonos_tts_utils.py,sha256=bwu88wsJGzEqbssgb-wpS_7lFscJ74J8cgyca-hX_Qw,6422
 sinapsis_zonos/src/sinapsis_zonos/templates/__init__.py,sha256=A-_F0K3hbEFqeWWAh4YftgU9CFX-WHrauSiCAww9yp8,482
-sinapsis_zonos/src/sinapsis_zonos/templates/zonos_tts.py,sha256=iwdouNWVt1phgQdW1Z3XtDz18K9IRbcYLFTmkT0l6i4,7698
-sinapsis_speech-0.4.1.dist-info/METADATA,sha256=Kw8KxfiyvVXgIqv7T1HVytlk6DKe1JY0Tos7iSXjbjQ,12783
-sinapsis_speech-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sinapsis_speech-0.4.1.dist-info/top_level.txt,sha256=KvdwXupt5wnqb_4XGRcuJaL9Glgdw-DBvRkNzhgl_Ds,110
-sinapsis_speech-0.4.1.dist-info/RECORD,,
+sinapsis_zonos/src/sinapsis_zonos/templates/zonos_tts.py,sha256=h5EToXoJgAgjqvz9WLDfSjhCsV5zgBwZrX5cTJ4VnhM,7679
+sinapsis_speech-0.4.3.dist-info/METADATA,sha256=UZXiSYdXx8deyu9p28aaUkIDyxtfm1REJYrvV2JSj6E,12783
+sinapsis_speech-0.4.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sinapsis_speech-0.4.3.dist-info/top_level.txt,sha256=KvdwXupt5wnqb_4XGRcuJaL9Glgdw-DBvRkNzhgl_Ds,110
+sinapsis_speech-0.4.3.dist-info/RECORD,,

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."""