Fix empty latents bug from previous commit

2025-08-04 23:16:25 +03:00
parent fe7e124329
commit fc7f504e85
1 changed files with 2 additions and 4 deletions
@@ -932,11 +932,9 @@ class WanVideoEmptyEmbeds:
            "target_shape": target_shape,
            "num_frames": num_frames,
            "control_embeds": control_embeds["control_embeds"] if control_embeds is not None else None,
-            "extra_latents": [{
-                "samples": extra_latents["samples"],
-                "index": 0,
-            }]
        }
+        if extra_latents is not None:
+            embeds["extra_latents"] = extra_latents["samples"].squeeze(0)
    
        return (embeds,)