From 80e88b4f4dc9b90dab9bff0b3b81e031ea76523c Mon Sep 17 00:00:00 2001
From: Goekdeniz-Guelmez <Gulmezdeniz1999@gmail.com>
Date: Tue, 10 Dec 2024 18:18:59 +0100
Subject: [PATCH] nits

---
 llms/mlx_lm/models/mamba2.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/llms/mlx_lm/models/mamba2.py b/llms/mlx_lm/models/mamba2.py
index 981daa74..822ebe8a 100644
--- a/llms/mlx_lm/models/mamba2.py
+++ b/llms/mlx_lm/models/mamba2.py
@@ -65,6 +65,7 @@ def silu(x):
 
 def ssd(x, A, B, C, chunk_size):
     batch, seqlen, nheads, dim = x.shape
+
     B = mx.expand_dims(B, axis=2)
     C = mx.expand_dims(C, axis=2)