From 55a555850d846d59f7a448b2082e390a7cd6f9f3 Mon Sep 17 00:00:00 2001
From: Anthony Mercurio <seizou.genkei@protonmail.com>
Date: Wed, 30 Nov 2022 22:52:48 -0700
Subject: [PATCH] Remove dangling barrier

---
 trainer/diffusers_trainer.py | 1 -
 1 file changed, 1 deletion(-)

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())