Support for Hunyuan Video Avatar

2025-06-05 15:35:58 +02:00
parent 7670af9610
commit 1976868f6a
15 changed files with 2562 additions and 81 deletions
--- a/hyvideo/modules/embed_layers.py
+++ b/hyvideo/modules/embed_layers.py
@@ -53,10 +53,11 @@ class PatchEmbed(nn.Module):

    def forward(self, x):
        x = self.proj(x)
+        shape = x.shape
        if self.flatten:
            x = x.flatten(2).transpose(1, 2)  # BCHW -> BNC
        x = self.norm(x)
-        return x
+        return x, shape


 class TextProjection(nn.Module):