Automated PR - 2026-03-04

2026-03-04 19:34:46 +00:00
parent 28c3c73fe5
commit 822ce3c4b1
73 changed files with 4984 additions and 1220 deletions
@@ -67,14 +67,18 @@ class DummyDataset(Dataset):
                "fps": self.fps,
            },
            "text_conditions": {
-                "prompt_embeds": torch.randn(
+                "video_prompt_embeds": torch.randn(
                    self.prompt_sequence_length,
                    self.prompt_embed_dim,
-                ),  # random text embeddings
+                ),
+                "audio_prompt_embeds": torch.randn(
+                    self.prompt_sequence_length,
+                    self.prompt_embed_dim,
+                ),
                "prompt_attention_mask": torch.ones(
                    self.prompt_sequence_length,
                    dtype=torch.bool,
-                ),  # random attention mask
+                ),
            },
        }