Hugging Face's logo

kernels-community
/
mamba-ssm

kernel
Model card Files
xet
Community
3
medmekk HF Staff commited on
Commit
8ce5a49
·
1 Parent(s): 3f2a020

Fix for Falcon

Browse files
Files changed (6) hide show
  1. build/torch28-cxx11-cu126-x86_64-linux/__init__.py +2 -1
  2. build/torch28-cxx11-cu128-x86_64-linux/__init__.py +2 -1
  3. build/torch28-cxx11-cu129-x86_64-linux/__init__.py +2 -1
  4. build/torch29-cxx11-cu126-x86_64-linux/__init__.py +2 -1
  5. build/torch29-cxx11-cu128-x86_64-linux/__init__.py +2 -1
  6. build/torch29-cxx11-cu130-x86_64-linux/__init__.py +2 -1
build/torch28-cxx11-cu126-x86_64-linux/__init__.py CHANGED
@@ -5,7 +5,8 @@ from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
 
8
  __all__ = [
9
- "selective_scan_fn", "mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
10
  "Mamba", "Mamba2", "MambaLMHeadModel",
11
  ]
 
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
8
+ falcon_mamba_inner_fn = mamba_inner_fn
9
  __all__ = [
10
+ "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
11
  "Mamba", "Mamba2", "MambaLMHeadModel",
12
  ]
build/torch28-cxx11-cu128-x86_64-linux/__init__.py CHANGED
@@ -5,7 +5,8 @@ from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
 
8
  __all__ = [
9
- "selective_scan_fn", "mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
10
  "Mamba", "Mamba2", "MambaLMHeadModel",
11
  ]
 
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
8
+ falcon_mamba_inner_fn = mamba_inner_fn
9
  __all__ = [
10
+ "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
11
  "Mamba", "Mamba2", "MambaLMHeadModel",
12
  ]
build/torch28-cxx11-cu129-x86_64-linux/__init__.py CHANGED
@@ -5,7 +5,8 @@ from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
 
8
  __all__ = [
9
- "selective_scan_fn", "mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
10
  "Mamba", "Mamba2", "MambaLMHeadModel",
11
  ]
 
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
8
+ falcon_mamba_inner_fn = mamba_inner_fn
9
  __all__ = [
10
+ "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
11
  "Mamba", "Mamba2", "MambaLMHeadModel",
12
  ]
build/torch29-cxx11-cu126-x86_64-linux/__init__.py CHANGED
@@ -5,7 +5,8 @@ from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
 
8
  __all__ = [
9
- "selective_scan_fn", "mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
10
  "Mamba", "Mamba2", "MambaLMHeadModel",
11
  ]
 
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
8
+ falcon_mamba_inner_fn = mamba_inner_fn
9
  __all__ = [
10
+ "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
11
  "Mamba", "Mamba2", "MambaLMHeadModel",
12
  ]
build/torch29-cxx11-cu128-x86_64-linux/__init__.py CHANGED
@@ -5,7 +5,8 @@ from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
 
8
  __all__ = [
9
- "selective_scan_fn", "mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
10
  "Mamba", "Mamba2", "MambaLMHeadModel",
11
  ]
 
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
8
+ falcon_mamba_inner_fn = mamba_inner_fn
9
  __all__ = [
10
+ "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
11
  "Mamba", "Mamba2", "MambaLMHeadModel",
12
  ]
build/torch29-cxx11-cu130-x86_64-linux/__init__.py CHANGED
@@ -5,7 +5,8 @@ from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
 
8
  __all__ = [
9
- "selective_scan_fn", "mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
10
  "Mamba", "Mamba2", "MambaLMHeadModel",
11
  ]
 
5
  from .modules import Mamba, Mamba2
6
  from .models import MambaLMHeadModel
7
 
8
+ falcon_mamba_inner_fn = mamba_inner_fn
9
  __all__ = [
10
+ "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
11
  "Mamba", "Mamba2", "MambaLMHeadModel",
12
  ]