export the sigmoid contrastive learning module

Author: lucidrains

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

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',