Method __init__

diffsynth/models/cog_dit.py:12–14

(self, dim_in, dim_out, patch_size)

class CogPatchify(torch.nn.Module):
    def __init__(self, dim_in, dim_out, patch_size) -> None:
        super().__init__()
        self.proj = torch.nn.Conv3d(dim_in, dim_out, kernel_size=(1, patch_size, patch_size), stride=(1, patch_size, patch_size))

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