code: def __init__(
      self,
      embedding_dim: int,
      mlp_dim: int,
      act: type[Module] = nn.GELU,
  ) -> None:
      super().__init__()
      self.lin1 = nn.Linear(embedding_dim, mlp_dim)
      self.lin2 = nn.Linear(mlp_dim, embedding_dim)
      self.act = act()
  
file_path: kornia/kornia/contrib/models/sam/architecture/common.py
imports: from __future__ import annotations
  from typing import Any
  from torch import nn
  from kornia.core import Module, Tensor
label_desc: Use image node_id 1 to create a new MLPBlock object from inherited base classes: Module with example: obj = MLPBlock(embedding_dim, mlp_dim, act)