revert assertion change for now

EleutherAI · Aug 24, 2024 · 94b03f5 · 94b03f5
1 parent 0aaaa32
commit 94b03f5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/megatron/model/mamba/mamba.py b/megatron/model/mamba/mamba.py
@@ -13,8 +13,9 @@
     from causal_conv1d import causal_conv1d_fn
     import einops
 except ModuleNotFoundError:
-    assert False, "Unable to import Mamba kernels. Install them from our requirements/requirements-mamba.txt, \
-    or directly from https://github.com/state-spaces/mamba"
+    print( "Unable to import Mamba kernels. Install them from our requirements/requirements-mamba.txt, \
+    or directly from https://github.com/state-spaces/mamba")
+    pass
 
 from megatron.model.norms import get_norm
 from megatron import mpu