Assertion Error: world_size % group_num == 0 When finetune cogv1.5-i2v on distributed environment

[hrcheng98]

System Info / 系統信息

Information / 问题信息

• [ ] The official example scripts / 官方的示例脚本
• [X] My own modified scripts / 我自己修改的脚本和任务

Reproduction / 复现过程

1. run code on 2*8 GPU
2. find assertion error world_size % group_num != 0

Expected behavior / 期待表现

I found group_num in loss_fn_config in yaml is 40. Should the group_num be modified according to the machine amount I am using?