Method __init__

k_diffusion/models/modules.py:610–615

(self, in_features, out_features, patch_size=(2, 2))

    
class TokenSplitWithoutSkip(nn.Module):
    def __init__(self, in_features, out_features, patch_size=(2, 2)):
        super().__init__()
        self.h = patch_size[0]
        self.w = patch_size[1]
        self.proj = apply_wd(Linear(in_features, out_features * self.h * self.w, bias=False))
        nn.init.zeros_(self.proj.weight)

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