From 097c6f46c68c607fd4f042d572304d39b8495cf5 Mon Sep 17 00:00:00 2001 From: AFThielmann Date: Tue, 12 Nov 2024 22:32:10 +0100 Subject: [PATCH] original_mamba dt_rank fix --- mambular/arch_utils/mamba_utils/mamba_original.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/mambular/arch_utils/mamba_utils/mamba_original.py b/mambular/arch_utils/mamba_utils/mamba_original.py index b5e726a..988abdc 100644 --- a/mambular/arch_utils/mamba_utils/mamba_original.py +++ b/mambular/arch_utils/mamba_utils/mamba_original.py @@ -77,7 +77,6 @@ def __init__( d_state=d_state, d_conv=d_conv, expand=expand_factor, - dt_rank=dt_rank, dt_min=dt_min, dt_max=dt_max, dt_init=dt_init, @@ -85,7 +84,6 @@ def __init__( dt_init_floor=dt_init_floor, conv_bias=conv_bias, bias=bias, - use_fast_path=True, # Fused kernel options layer_idx=layer_idx, ) self.norm = norm