Method __init__

diffsynth/models/hunyuan_video_dit.py:197–199

(self, patch_size=(1, 2, 2), in_channels=16, embed_dim=3072)

class PatchEmbed(torch.nn.Module):
    def __init__(self, patch_size=(1, 2, 2), in_channels=16, embed_dim=3072):
        super().__init__()
        self.proj = torch.nn.Conv3d(in_channels, embed_dim, kernel_size=patch_size, stride=patch_size)

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