From ea38b9583ab6d8cee78b11301489aaf097b4f7a5 Mon Sep 17 00:00:00 2001
From: jungwon-choi <jungwon.choi@kaist.ac.kr>
Date: Thu, 2 May 2024 06:56:01 +0900
Subject: [PATCH] Fix wrong projection output

---
 vision_mamba/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vision_mamba/model.py b/vision_mamba/model.py
index ccd6fac..cb7af7c 100644
--- a/vision_mamba/model.py
+++ b/vision_mamba/model.py
@@ -95,7 +95,7 @@ def forward(self, x: torch.Tensor):
 
         # Split x into x1 and x2 with linears
         z1 = self.proj(x)
-        x1 = self.proj(x)
+        x = self.proj(x)
 
         # forward con1d
         x1 = self.process_direction(