daniel/h3-blackwell-runtime
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Match upstream VAE temporal padding

Browse source
This commit is contained in:
Daniel Maddern 2026-08-14 01:25:53 +07:00
parent 6dd0b0e737
commit 49bae4ab74
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/h3_blackwell_runtime/vae_decoder.py
View file

@ -316,7 +316,7 @@ class MiniMaxH3VideoVAE(nn.Module):
pad_tokens += self.tokens_chunk_size
chunks += 1
if pad_tokens:
z = torch.cat((z, z[:, :, -1:].expand(-1, -1, pad_tokens, -1, -1)), dim=2)
z = torch.cat((z, z[:, :, -1:].repeat(1, 1, pad_tokens, 1, 1)), dim=2)
output_frames = self._decode_temporal_frame_plan(z.shape[2], chunks, pad_tokens)
output = None
overlap = None