From c00340c10756c5bc5c44f0ac24bfdfcaa9bfcaaa Mon Sep 17 00:00:00 2001 From: Xlucidator Date: Wed, 17 Dec 2025 14:25:13 +0800 Subject: [PATCH] [fix] clear typo slip of 'if_divide_out' in model registration. --- vim/models_mamba.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vim/models_mamba.py b/vim/models_mamba.py index 3be8bb15..b7314fe4 100644 --- a/vim/models_mamba.py +++ b/vim/models_mamba.py @@ -607,7 +607,7 @@ def vim_small_patch16_stride8_224_bimambav2_final_pool_mean_abs_pos_embed_with_m @register_model def vim_base_patch16_224_bimambav2_final_pool_mean_abs_pos_embed_with_middle_cls_token_div2(pretrained=False, **kwargs): model = VisionMamba( - patch_size=16, embed_dim=768, d_state=16, depth=24, rms_norm=True, residual_in_fp32=True, fused_add_norm=True, final_pool_type='mean', if_abs_pos_embed=True, if_rope=False, if_rope_residual=False, bimamba_type="v2", if_cls_token=True, if_devide_out=True, use_middle_cls_token=True, **kwargs) + patch_size=16, embed_dim=768, d_state=16, depth=24, rms_norm=True, residual_in_fp32=True, fused_add_norm=True, final_pool_type='mean', if_abs_pos_embed=True, if_rope=False, if_rope_residual=False, bimamba_type="v2", if_cls_token=True, if_divide_out=True, use_middle_cls_token=True, **kwargs) model.default_cfg = _cfg() if pretrained: checkpoint = torch.hub.load_state_dict_from_url(