Method __init__

k_diffusion/models/modules.py:129–135

(self, in_features, out_features, kernel_size, bias=True)

class ConvGEGLU(nn.Module):
    def __init__(self, in_features, out_features, kernel_size, bias=True):
        super().__init__()
        self.conv = torch.nn.Conv2d(in_features,
                                 out_features*2,
                                 kernel_size=kernel_size,
                                 padding=kernel_size//2,
                                 bias=bias)

    def forward(self, x):
        x=self.conv(x)