nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-14 18:25:35 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

debug

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-19 10:51:11 +08:00
parent 41a600b61b
commit a78ffe95a6
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
scripts/inference-vae-v2.py
View file

@ -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)