1 files changed, 1 insertions, 1 deletions

diff --git a/train_ti.py b/train_ti.py
index 1054a5d..e6c437e 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -974,7 +974,7 @@ def main():
 
     try:
         for epoch in range(num_epochs):
-            if accelerator.is_main_process and False:
+            if accelerator.is_main_process:
                 if epoch % args.sample_frequency == 0:
                     checkpointer.save_samples(global_step + global_step_offset, args.sample_steps)