BonanDing commited on
Commit
270eca8
·
1 Parent(s): 70c492d

Remove DeMemWM df_base debugger trap

Browse files
Files changed (1) hide show
  1. algorithms/dememwm/df_base.py +1 -4
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
- try:
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(