Bugfix: fix memory allocation issues during multi-GPU training#67

LittleNyima wants to merge 1 commit intoLightricks:mainLightricks/LTX-2:mainfrom

LittleNyima:mainLittleNyima/LTX-2:mainCopy head branch name to clipboard