JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-05-16 16:49:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update diffusion_video.py

Browse Source
This commit is contained in:
zR 2025-01-03 08:45:42 +08:00
parent b080c6a010
commit ce2c299c1f
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
sat/diffusion_video.py
View File

@ -10,7 +10,6 @@ import torch
from torch import nn from torch import nn
from sgm.modules import UNCONDITIONAL_CONFIG from sgm.modules import UNCONDITIONAL_CONFIG
from sgm.modules.autoencoding.temporal_ae import VideoDecoder
from sgm.modules.diffusionmodules.wrappers import OPENAIUNETWRAPPER from sgm.modules.diffusionmodules.wrappers import OPENAIUNETWRAPPER
from sgm.util import ( from sgm.util import (
default, default,
@ -192,11 +191,7 @@ class SATVideoDiffusionEngine(nn.Module):
for n in range(n_rounds): for n in range(n_rounds):
z_now = z[n * n_samples : (n + 1) * n_samples, :, 1:] z_now = z[n * n_samples : (n + 1) * n_samples, :, 1:]
latent_time = z_now.shape[2] # check the time latent latent_time = z_now.shape[2] # check the time latent
temporal_compress_times = 4
fake_cp_size = min(10, latent_time // 2) fake_cp_size = min(10, latent_time // 2)
start_frame = 0
recons = [] recons = []
start_frame = 0 start_frame = 0
for i in range(fake_cp_size): for i in range(fake_cp_size):