Fix transition decode: flatten [B, T, H, W, C] to [N, H, W, C]
LTXAV video VAE returns 5D tensor; _blur_pixel_frame expected 3D frames. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Sonnet 4.6
parent
9dbd239702
commit
a46c4c7146
+6
-1
@@ -1274,11 +1274,16 @@ class FunPackLTXAVSceneChainSampler:
|
||||
|
||||
window = self._time_slice(video_tensor, start_l, end_l)
|
||||
try:
|
||||
decoded = vae.decode(window) # [N, H, W, C]
|
||||
decoded = vae.decode(window)
|
||||
except Exception as exc:
|
||||
print(f"[FunPackSceneChain] {effect} decode failed: {exc}")
|
||||
continue
|
||||
|
||||
# Video VAEs may return [B, T, H, W, C] — flatten to [N, H, W, C]
|
||||
if decoded.dim() == 5:
|
||||
b, t, h, w, c = decoded.shape
|
||||
decoded = decoded.reshape(b * t, h, w, c)
|
||||
|
||||
decoded = decoded.clone().float()
|
||||
n = decoded.shape[0]
|
||||
center = n // 2
|
||||
|
||||
Reference in New Issue
Block a user