From 94b03f5e1e5f7d506f7230724ad6caa6c6b866b2 Mon Sep 17 00:00:00 2001 From: Jacob Hatef Date: Sat, 24 Aug 2024 18:27:38 -0400 Subject: [PATCH] revert assertion change for now --- megatron/model/mamba/mamba.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/megatron/model/mamba/mamba.py b/megatron/model/mamba/mamba.py index 6d01e4822..3177267cb 100644 --- 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