From 9d1dfe8dbe16c432b951555a947191d2a75b47a2 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Fri, 13 Sep 2024 18:21:05 -0700
Subject: [PATCH] Incorrectly named head_hidden_size

---
 timm/models/mambaout.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timm/models/mambaout.py b/timm/models/mambaout.py
index a33554a9..c2f2f07b 100644
--- a/timm/models/mambaout.py
+++ b/timm/models/mambaout.py
@@ -384,7 +384,7 @@ class MambaOut(nn.Module):
                 drop_rate=drop_rate,
             )
         self.num_features = prev_dim
-        self.hidden_size = self.head.num_features
+        self.head_hidden_size = self.head.num_features
 
         self.apply(self._init_weights)