Method forward

k_diffusion/models/modules.py:137–144

(self, x)

                                 bias=bias)

    def forward(self, x):
        x=self.conv(x)
        # return linear_geglu(x, self.weight, self.bias)

        x, gate = x.chunk(2, dim=1)
        return x * F.gelu(gate)

        return x


# Layers