update

algorithms/worldmem/df_video.py

@@ -900,7 +900,7 @@ class WorldMemMinecraft(DiffusionForcingBase):
 
         xs_pred = self.decode(xs_pred[n_context_frames:].to(device)).cpu()
 
-        return xs_pred[-1,0], self_frames, self_poses, self_memory_c2w, self_frame_idx
+        return xs_pred[-1,0], self_frames, self_poses.cpu(), self_memory_c2w.cpu(), self_frame_idx.cpu()
 
 
     def reset(self):

app.py

@@ -277,8 +277,6 @@ def reset():
                                 self_memory_c2w=self_memory_c2w,
                                 self_frame_idx=self_frame_idx)
     
-
-    print("outoutout")
     return input_history, DEFAULT_IMAGE
 
 def on_image_click(SELECTED_IMAGE):