From ee4ce3f4a8ca7e12fc21d52d5adcaa5bbef90f14 Mon Sep 17 00:00:00 2001
From: Xuweiyi Chen <62165079+XuweiyiChen@users.noreply.github.com>
Date: Thu, 22 Feb 2024 23:39:41 -0500
Subject: [PATCH] Update util.py

---
 animatediff/utils/util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/animatediff/utils/util.py b/animatediff/utils/util.py
index c0944837..885a5de8 100644
--- a/animatediff/utils/util.py
+++ b/animatediff/utils/util.py
@@ -15,7 +15,7 @@
 
 
 def zero_rank_print(s):
-    if (not dist.is_initialized()) and (dist.is_initialized() and dist.get_rank() == 0): print("### " + s)
+    if (not dist.is_initialized()) or (dist.is_initialized() and dist.get_rank() == 0): print("### " + s)
 
 
 def save_videos_grid(videos: torch.Tensor, path: str, rescale=False, n_rows=6, fps=8):