update

algorithms/worldmem/df_video.py

@@ -797,16 +797,13 @@ class WorldMemMinecraft(DiffusionForcingBase):
         condition_similar_length = self.condition_similar_length
 
         if self_frames is None:
-            print("inside1", device)
             first_frame_encode = self.encode(first_frame[None, None].to(device))
             self_frames = first_frame_encode.cpu()
-            print("inside2", device)
             self.actions = curr_actions[None, None].to(device)
             self_poses = first_pose[None, None].to(device)
             new_c2w_mat = euler_to_camera_to_world_matrix(first_pose)
             self_memory_c2w = new_c2w_mat[None, None].to(device)
             self_frame_idx = torch.tensor([[context_frames_idx]]).to(device)
-            print("inside3")
             return first_frame, self_frames, self_poses, self_memory_c2w, self_frame_idx
         else:
             last_frame = self_frames[-1].clone()

app.py

@@ -278,7 +278,7 @@ def reset():
                                 self_frame_idx=self_frame_idx)
     
 
-
+    print("outoutout")
     return input_history, DEFAULT_IMAGE
 
 def on_image_click(SELECTED_IMAGE):