关于v2v_edit任务的多卡问题

作者您好！
我最近在使用您的模型进行v2v_edit任务，主要是把输入的视频中的某些主体去除。
但是由于我对视频的分辨率要求比较高，所以我设置的是1920*1080，81帧。
在这个情况下A100 80GB会OOM。由于我有四张这样的显卡，我在看过之前的一个关于分卡的issue后尝试按照您说的方法进行分卡，但是可能是我的方法有问题，分卡后数据似乎不能顺利流通，产出的结果是黑白马赛克视频。
因为我是一个刚刚入门这个领域的master program student，我怀疑是我对您的简短指示理解的不够透彻，可否烦请您详细一些地指教一下如何分卡？
我原本的分卡是在univideo_inference.py下进行的，具体来说是在transformer初始化之后，pipeline初始化之前把transformer的所有层平分给四个卡。
感谢！