Incorrectly named head_hidden_size

huggingface · Sep 14, 2024 · b0cfd9d · b0cfd9d
1 parent bab52f4
commit b0cfd9d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/timm/models/mambaout.py b/timm/models/mambaout.py
@@ -384,7 +384,7 @@ def __init__(
                 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)