diff --git a/trainer/diffusers_trainer.py b/trainer/diffusers_trainer.py
index 781b3f1..2b044bd 100644
--- a/trainer/diffusers_trainer.py
+++ b/trainer/diffusers_trainer.py
@@ -957,7 +957,6 @@ def main():
 
     save_checkpoint(global_step)
 
-    torch.distributed.barrier()
     cleanup()
 
     print(get_gpu_ram())