Inference Effor

Thank you for your excellent work. I found that the generated video is always wrong when I was inferring. I don't know if I did not handle it properly. The following example is inferred using asserts/example_images/3.png.
https://github.com/user-attachments/assets/6b672db6-e633-4d9a-be69-c184d0179242