ok it is working

lucidrains · Feb 2, 2023 · 26c8268 · 26c8268
1 parent 2de8bb0
commit 26c8268
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -95,7 +95,8 @@ wav2vec = HubertWithKmeans(
 semantic_transformer = SemanticTransformer(
     num_semantic_tokens = wav2vec.codebook_size,
     dim = 1024,
-    depth = 6
+    depth = 6,
+    audio_text_condition = True      # this must be set to True (same for CoarseTransformer and FineTransformers)
 ).cuda()
 
 trainer = SemanticTransformerTrainer(

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'musiclm-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.7',
+  version = '0.0.8',
   license='MIT',
   description = 'MusicLM - AudioLM + Audio CLIP to text to music synthesis',
   author = 'Phil Wang',
@@ -19,7 +19,7 @@
     'contrastive learning'
   ],
   install_requires=[
-    'audiolm-pytorch>=0.9.2',
+    'audiolm-pytorch>=0.9.3',
     'beartype',
     'einops>=0.4',
     'vector-quantize-pytorch>=1.0.0',