From d40b7d4d1d07a1c4932e298c26c8880beb389289 Mon Sep 17 00:00:00 2001 From: lucidrains Date: Sat, 27 Jul 2024 06:49:59 -0700 Subject: [PATCH] default sampling steps to 32 --- e2_tts_pytorch/e2_tts.py | 4 ++-- pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/e2_tts_pytorch/e2_tts.py b/e2_tts_pytorch/e2_tts.py index cb25526..ba9b32e 100644 --- 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 index fe2d17e..084594b 100644 --- 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 = "lucidrains@gmail.com" }