Commit ·
d4b4d20
1
Parent(s): 383d55e
fix
Browse files- __pycache__/model.cpython-312.pyc +0 -0
- model.py +2 -2
__pycache__/model.cpython-312.pyc
CHANGED
|
Binary files a/__pycache__/model.cpython-312.pyc and b/__pycache__/model.cpython-312.pyc differ
|
|
|
model.py
CHANGED
|
@@ -9,8 +9,8 @@ from causal_conv1d.causal_conv1d_varlen import causal_conv1d_varlen_states
|
|
| 9 |
from mamba_ssm.ops.triton.selective_state_update import selective_state_update
|
| 10 |
|
| 11 |
# --- TODO: These two are always compiled even when kernel.compile is disabled. We should fix this. ---
|
| 12 |
-
from causal_conv1d_compilable import causal_conv1d_fn, causal_conv1d_update
|
| 13 |
-
from ssm_compilable import mamba_chunk_scan_combined
|
| 14 |
# -----------------------------------------------------------------------------------------------------
|
| 15 |
|
| 16 |
from .norms import build_norm
|
|
|
|
| 9 |
from mamba_ssm.ops.triton.selective_state_update import selective_state_update
|
| 10 |
|
| 11 |
# --- TODO: These two are always compiled even when kernel.compile is disabled. We should fix this. ---
|
| 12 |
+
from .causal_conv1d_compilable import causal_conv1d_fn, causal_conv1d_update
|
| 13 |
+
from .ssm_compilable import mamba_chunk_scan_combined
|
| 14 |
# -----------------------------------------------------------------------------------------------------
|
| 15 |
|
| 16 |
from .norms import build_norm
|