diff --git a/naturalspeech2_pytorch/naturalspeech2_pytorch.py b/naturalspeech2_pytorch/naturalspeech2_pytorch.py index 8e2beae..cef9344 100644 --- a/naturalspeech2_pytorch/naturalspeech2_pytorch.py +++ b/naturalspeech2_pytorch/naturalspeech2_pytorch.py @@ -1074,7 +1074,7 @@ def process_prompt(self, prompt = None): if is_raw_prompt: with torch.no_grad(): self.codec.eval() - prompt, _, _ = self.codec(prompt, return_encoded = True) + prompt, _, _ = self.codec(prompt, curtail_from_left = True, return_encoded = True) return prompt diff --git a/naturalspeech2_pytorch/version.py b/naturalspeech2_pytorch/version.py index 858d0a1..af2cb9a 100644 --- a/naturalspeech2_pytorch/version.py +++ b/naturalspeech2_pytorch/version.py @@ -1 +1 @@ -__version__ = '0.0.25' +__version__ = '0.0.26' diff --git a/setup.py b/setup.py index 69caed4..48c640b 100644 --- a/setup.py +++ b/setup.py @@ -20,7 +20,7 @@ ], install_requires=[ 'accelerate', - 'audiolm-pytorch>=0.30.0', + 'audiolm-pytorch>=0.30.2', 'beartype', 'einops>=0.6.1', 'ema-pytorch',