From a8205b575dad5fcb86ccd094a032223fa5e34da3 Mon Sep 17 00:00:00 2001
From: zR <2448370773@qq.com>
Date: Fri, 8 Nov 2024 23:27:44 +0800
Subject: [PATCH] Update cp_enc_dec.py

---
 sat/vae_modules/cp_enc_dec.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/sat/vae_modules/cp_enc_dec.py b/sat/vae_modules/cp_enc_dec.py
index 094f5ae..4d77324 100644
--- a/sat/vae_modules/cp_enc_dec.py
+++ b/sat/vae_modules/cp_enc_dec.py
@@ -517,11 +517,8 @@ class Upsample3D(nn.Module):
     def forward(self, x, fake_cp=True):
         if self.compress_time and x.shape[2] > 1:
             if get_context_parallel_rank() == 0 and fake_cp:
-                print(x.shape)
-                breakpoint()
                 # split first frame
                 x_first, x_rest = x[:, :, 0], x[:, :, 1:]
-
                 x_first = torch.nn.functional.interpolate(x_first, scale_factor=2.0, mode="nearest")
 
                 splits = torch.split(x_rest, 32, dim=1)