curtail from the left when the prompt is being encoded, only works fo…

…r soundstream for now
lucidrains · May 13, 2023 · 105d2a4 · 105d2a4
1 parent e1c4667
commit 105d2a4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -1 +1 @@
-__version__ = '0.0.25'
+__version__ = '0.0.26'
diff --git 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',