From a78ffe95a6450640aa5b6278695f402bcfcb1b8c Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Fri, 19 Apr 2024 10:51:11 +0800 Subject: [PATCH] debug --- scripts/inference-vae-v2.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/scripts/inference-vae-v2.py b/scripts/inference-vae-v2.py index 4ccab91..4c743b8 100644 --- a/scripts/inference-vae-v2.py +++ b/scripts/inference-vae-v2.py @@ -145,7 +145,10 @@ def main(): loss_steps = 0 disc_time_downsample_factor = 2 ** len(cfg.discriminator.channel_multipliers) - disc_time_padding = disc_time_downsample_factor - cfg.num_frames % disc_time_downsample_factor + if cfg.num_frames % disc_time_downsample_factor != 0: + disc_time_padding = disc_time_downsample_factor - cfg.num_frames % disc_time_downsample_factor + else: + disc_time_padding = 0 video_contains_first_frame = cfg.video_contains_first_frame lecam_ema_real = torch.tensor(0.0)