bug fix

FeixLiu · Jul 8, 2024 · e5eeaa0 · e5eeaa0
1 parent bdd2287
commit e5eeaa0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlenlp/utils/distributed.py b/paddlenlp/utils/distributed.py
@@ -214,7 +214,7 @@ def distributed_allgather(tensor: Any, group=None, offload=False):
                 x.reshape_(origin_shape)
 
         else:
-            distributed.all_gather(output_tensors, tensor)
+            distributed.all_gather(output_tensors, tensor, group=group)
 
         return output_tensors