default sampling steps to 32

lucidrains · Jul 27, 2024 · d40b7d4 · d40b7d4
1 parent cb56ac6
commit d40b7d4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/e2_tts_pytorch/e2_tts.py b/e2_tts_pytorch/e2_tts.py
@@ -585,8 +585,8 @@ def sample(
         text: Int['b n'] | List[str] | None = None,
         lens: Int['b'] | None = None,
         duration: int | Int['b'] | None = None,
-        steps = 3,
-        cfg_strength = 1.,   # they used a classifier free guidance strenght of 1.
+        steps = 32,
+        cfg_strength = 1.,   # they used a classifier free guidance strength of 1.
         max_duration = 4096, # in case the duration predictor goes haywire
         vocoder: Callable[Float['b d n'], Float['b nw']] | None = None
     ):

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "e2-tts-pytorch"
-version = "0.1.9"
+version = "0.1.10"
 description = "E2-TTS in Pytorch"
 authors = [
     { name = "Phil Wang", email = "[email protected]" }