wan_video_dit.py中的patchify函数是错误实现

<img width="1197" height="689" alt="Image" src="https://github.com/user-attachments/assets/a1d173dc-0f57-4828-a57b-27acbc34a3fa" />

主要有2个错误：
1. 返回参数的数量不同，patchify返回x，而接受是x, grid_size(含f, h, w)
2. patchify返回的x的shape不正确