def init_weights_vit_timm()

in deepseek_vl2/models/siglip_vit.py [0:0]

• 7 lines of code
• 4 McCabe index (conditional complexity)

def init_weights_vit_timm(module: nn.Module, name: str = '') -> None:
    """ ViT weight initialization, original timm impl (for reproducibility) """
    if isinstance(module, nn.Linear):
        trunc_normal_(module.weight, std=.02)
        if module.bias is not None:
            nn.init.zeros_(module.bias)
    elif hasattr(module, 'init_weights'):
        module.init_weights()