Method __init__

encoders/dinov2/layers/block.py:126–127

(self, *args, **kwargs)

# Override forward with modulation input
class BlockWithModulation(Block):
    def __init__(self, *args, **kwargs) -> None:
        super().__init__(*args, **kwargs)

    def forward(self, x: Tensor, mod: Tensor) -> Tensor:
        def attn_residual_func(x: Tensor, mod: Tensor) -> Tensor: