Remove DeMemWM df_base debugger trap
Browse files
algorithms/dememwm/df_base.py
CHANGED
|
@@ -163,10 +163,7 @@ class DiffusionForcingBase(BasePytorchAlgo):
|
|
| 163 |
# update xs_pred by DDIM or DDPM sampling
|
| 164 |
# input frames within the sliding window
|
| 165 |
|
| 166 |
-
|
| 167 |
-
input_condition = conditions[start_frame : curr_frame + horizon].clone()
|
| 168 |
-
except:
|
| 169 |
-
import pdb;pdb.set_trace()
|
| 170 |
if self.condtion_similar_length:
|
| 171 |
input_condition = torch.cat([conditions[start_frame : curr_frame + horizon], conditions[-self.condtion_similar_length:]], dim=0)
|
| 172 |
xs_pred[start_frame:] = self.diffusion_model.sample_step(
|
|
|
|
| 163 |
# update xs_pred by DDIM or DDPM sampling
|
| 164 |
# input frames within the sliding window
|
| 165 |
|
| 166 |
+
input_condition = conditions[start_frame : curr_frame + horizon].clone()
|
|
|
|
|
|
|
|
|
|
| 167 |
if self.condtion_similar_length:
|
| 168 |
input_condition = torch.cat([conditions[start_frame : curr_frame + horizon], conditions[-self.condtion_similar_length:]], dim=0)
|
| 169 |
xs_pred[start_frame:] = self.diffusion_model.sample_step(
|