我想挑一个已预训练的模型来跑我的4k视频的光流，跑了raft和gma但是总是有cuda out of memory的问题。请问有合适的模型可以跑4k视频吗？或者说cuda的问题可以解决吗？非常感谢🙏
试试RAFT呢