Attempted to set the storage of a tensor on device "cpu" to a storage on different device "mps:0"

**When i use mps or cpu to run the video-predictor code in notebook, everything about picture segment is ok, but when it comes to the video predictor, the code go wrong like this:**

<img width="954" alt="截屏2024-12-12 13 50 30" src="https://github.com/user-attachments/assets/e19e85c2-bf49-49e9-80b1-4ae9087fabed" />
sonthing meet same problem with me?
my device is mbp m4