JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-04-06 03:57:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

use original up.upsample

Browse Source
This commit is contained in:
zR 2024-11-05 22:06:05 +08:00
parent 4a3035d64e
commit 0ae12e3ea3
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
sat/vae_modules/cp_enc_dec.py
View File

@ -960,10 +960,10 @@ class ContextParallelDecoder3D(nn.Module):
up.upsample = Upsample3D(block_in, with_conv=resamp_with_conv, compress_time=True)
else:
up.upsample = Upsample3D(block_in, with_conv=resamp_with_conv, compress_time=False)
# if i_level < self.num_resolutions - self.temporal_compress_level:
# up.upsample = Upsample3D(block_in, with_conv=resamp_with_conv, compress_time=False)
# else:
# up.upsample = Upsample3D(block_in, with_conv=resamp_with_conv, compress_time=True)
if i_level < self.num_resolutions - self.temporal_compress_level:
up.upsample = Upsample3D(block_in, with_conv=resamp_with_conv, compress_time=False)
else:
up.upsample = Upsample3D(block_in, with_conv=resamp_with_conv, compress_time=True)
self.up.insert(0, up)
self.norm_out = Normalize3D(block_in, zq_ch, add_conv=add_conv, gather=gather_norm)