Skip to content

Commit a8205b5

Browse files
Update cp_enc_dec.py
1 parent e7bcecf commit a8205b5

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

sat/vae_modules/cp_enc_dec.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -517,11 +517,8 @@ def __init__(
517517
def forward(self, x, fake_cp=True):
518518
if self.compress_time and x.shape[2] > 1:
519519
if get_context_parallel_rank() == 0 and fake_cp:
520-
print(x.shape)
521-
breakpoint()
522520
# split first frame
523521
x_first, x_rest = x[:, :, 0], x[:, :, 1:]
524-
525522
x_first = torch.nn.functional.interpolate(x_first, scale_factor=2.0, mode="nearest")
526523

527524
splits = torch.split(x_rest, 32, dim=1)

0 commit comments

Comments
 (0)