quick fix

lucidrains · Sep 28, 2023 · 5c85f21 · 5c85f21
1 parent def5b44
commit 5c85f21
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'voicebox-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.1.7',
+  version = '0.1.8',
   license='MIT',
   description = 'Voicebox - Pytorch',
   author = 'Phil Wang',

diff --git a/voicebox_pytorch/voicebox_pytorch.py b/voicebox_pytorch/voicebox_pytorch.py
@@ -987,7 +987,8 @@ def sample(
         cond_mask = None,
         steps = 3,
         cond_scale = 1.,
-        decode_to_audio = True
+        decode_to_audio = True,
+        max_semantic_token_ids = 2048
     ):
         # take care of condition as raw audio
 
@@ -1018,7 +1019,7 @@ def sample(
                         source = default(text_token_ids, texts),
                         source_type = 'text',
                         target_type = 'speech',
-                        max_length = 10,
+                        max_length = max_semantic_token_ids,
                         return_target_mask = True
                     )