asdf98 commited on
Commit
a5069eb
·
verified ·
1 Parent(s): 2bf4fa5

Upload musemorphic/__init__.py

Browse files
Files changed (1) hide show
  1. musemorphic/__init__.py +3 -1
musemorphic/__init__.py CHANGED
@@ -2,7 +2,7 @@
2
  MuseMorphic: Lightweight Consumer-Grade MIDI Generation
3
  """
4
 
5
- __version__ = "0.1.0"
6
 
7
  from .model import (
8
  MuseMorphicConfig,
@@ -14,6 +14,7 @@ from .model import (
14
  SelectiveSSM,
15
  MambaBlock,
16
  ZClip,
 
17
  model_summary,
18
  )
19
 
@@ -34,6 +35,7 @@ __all__ = [
34
  'SelectiveSSM',
35
  'MambaBlock',
36
  'ZClip',
 
37
  'model_summary',
38
  'TokenizerConfig',
39
  'REMIPlusTokenizer',
 
2
  MuseMorphic: Lightweight Consumer-Grade MIDI Generation
3
  """
4
 
5
+ __version__ = "0.2.0"
6
 
7
  from .model import (
8
  MuseMorphicConfig,
 
14
  SelectiveSSM,
15
  MambaBlock,
16
  ZClip,
17
+ apply_span_mask_vectorized,
18
  model_summary,
19
  )
20
 
 
35
  'SelectiveSSM',
36
  'MambaBlock',
37
  'ZClip',
38
+ 'apply_span_mask_vectorized',
39
  'model_summary',
40
  'TokenizerConfig',
41
  'REMIPlusTokenizer',