From feeb4b75c90177d5f22ba07b9c6407ad9c8aec91 Mon Sep 17 00:00:00 2001 From: Daniel Maddern Date: Fri, 14 Aug 2026 01:50:14 +0700 Subject: [PATCH] Trace VAE blend inputs --- tools/compare_vae_temporal_assembly.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tools/compare_vae_temporal_assembly.py b/tools/compare_vae_temporal_assembly.py index 2643d7c..b655cc4 100644 --- a/tools/compare_vae_temporal_assembly.py +++ b/tools/compare_vae_temporal_assembly.py @@ -69,6 +69,9 @@ with torch.inference_mode(): part_d = clip_d[:, :, frame_start:frame_end][:, :, direct.frame_pre_padding:] part_u = clip_u[:, :, frame_start:frame_end][:, :, upstream.frame_pre_padding:] if split == 0 and prev_d is not None: + diff_stats(f"chunk_{chunk_index}_preblend_current", part_d, part_u) + diff_stats(f"chunk_{chunk_index}_direct_blend_same_inputs", direct.blend(prev_d, part_d, direct.frame_overlap, -3), upstream.blend(prev_d, part_d, direct.frame_overlap, -3)) + diff_stats(f"chunk_{chunk_index}_upstream_blend_same_inputs", direct.blend(prev_u, part_u, upstream.frame_overlap, -3), upstream.blend(prev_u, part_u, upstream.frame_overlap, -3)) part_d = direct.blend(prev_d, part_d, direct.frame_overlap, -3) part_u = upstream.blend(prev_u, part_u, upstream.frame_overlap, -3) diff_stats(f"chunk_{chunk_index}_blended_at_{write_pos}", part_d, part_u)