Add functions to export and import the model state for a given voice (#129)

2026-02-16 15:37:00 +01:00
parent aca7dc8db6
commit 8443de8e06
11 changed files with 123 additions and 212 deletions
@@ -6,7 +6,7 @@ from pocket_tts.models.tts_model import TTSModel as TTSModelImpl


 def test_public_api_exports_only_tts_model():
-    assert pocket_tts.__all__ == ["TTSModel"]
+    assert pocket_tts.__all__ == ["TTSModel", "export_model_state"]


 def test_public_api_tts_model_points_to_implementation():