export the sigmoid contrastive learning module

lucidrains · Sep 6, 2023 · 80a9826 · 80a9826
1 parent 90021a1
commit 80a9826
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/musiclm_pytorch/__init__.py b/musiclm_pytorch/__init__.py
@@ -1,5 +1,11 @@
-from musiclm_pytorch.musiclm_pytorch import MuLaN, MuLaNEmbedQuantizer, MusicLM
-
-from musiclm_pytorch.musiclm_pytorch import AudioSpectrogramTransformer, TextTransformer
+from musiclm_pytorch.musiclm_pytorch import (
+ MuLaN,
+ MuLaNEmbedQuantizer,
+ MusicLM,
+ AudioSpectrogramTransformer,
+ TextTransformer,
+ SigmoidContrastiveLearning,
+ SoftmaxContrastiveLearning
+)
 
 from musiclm_pytorch.trainer import MuLaNTrainer
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
  name = 'musiclm-pytorch',
  packages = find_packages(exclude=[]),
- version = '0.2.6',
+ version = '0.2.7',
  license='MIT',
  description = 'MusicLM - AudioLM + Audio CLIP to text to music synthesis',
  author = 'Phil Wang',