Method __init__

diffsynth/models/stepvideo_dit.py:571–574

(self, dim_in: int, dim_out: int, approximate: str = "none", bias: bool = True)

    """

    def __init__(self, dim_in: int, dim_out: int, approximate: str = "none", bias: bool = True):
        super().__init__()
        self.proj = nn.Linear(dim_in, dim_out, bias=bias)
        self.approximate = approximate

    def gelu(self, gate: torch.Tensor) -> torch.Tensor:
        return torch.nn.functional.gelu(gate, approximate=self.approximate)