Hugging Face's logo

Spaces:
P-rateek
/
MaIR
Runtime error

App Files Community
P-rateek commited on
Commit
eedf2f1
·
verified ·
1 Parent(s): b87079a

Update mamba_main/mamba_ssm/__init__.py

Browse files
Files changed (1) hide show
  1. mamba_main/mamba_ssm/__init__.py +3 -3
mamba_main/mamba_ssm/__init__.py CHANGED
@@ -1,5 +1,5 @@
1
  __version__ = "1.2.0.post1"
2
 
3
- from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
4
- from mamba_ssm.modules.mamba_simple import Mamba
5
- from mamba_ssm.models.mixer_seq_simple import MambaLMHeadModel
 
1
  __version__ = "1.2.0.post1"
2
 
3
+ from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
4
+ from .modules.mamba_simple import Mamba
5
+ from .models.mixer_seq_simple import MambaLMHeadModel