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

Fix VAE pixel normalization in-place bug

Browse source
This commit is contained in:
Daniel Maddern 2026-08-19 21:07:34 +07:00
parent 9bec53ac36
commit 390135fca6
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/h3_blackwell_runtime/vae_encoder.py
View file

@ -256,7 +256,8 @@ class MiniMaxH3VideoVAEEncoder(nn.Module):
"""``[B,3,H,W]`` or ``[B,3,T,H,W]`` pixels in ``[-1, 1]`` -> normalized latents ``[B,24,T_lat,H//16,W//16]``."""
if x.ndim == 4:
x = x.unsqueeze(2)
x = x.add(1.0).mul_(0.5).sub_(self.pixel_mean.to(x)).div_(self.pixel_std.to(x))
x = (x + 1.0) * 0.5
x = (x - self.pixel_mean.to(x)) / self.pixel_std.to(x)
if x.shape[2] == 1:
moments = self._adaptive_encode(x)[:, :, -1:, :, :]
else: