From a9cf49ecea8f0d5786c017b3fdb23dc475b5a313 Mon Sep 17 00:00:00 2001 From: glide-the Date: Wed, 27 Nov 2024 17:14:28 +0800 Subject: [PATCH] wandb tracker in scheduling problems during the training initiation and training stages. --- training/cogvideox_text_to_video_sft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/training/cogvideox_text_to_video_sft.py b/training/cogvideox_text_to_video_sft.py index ce4b6e0..ca50f69 100644 --- a/training/cogvideox_text_to_video_sft.py +++ b/training/cogvideox_text_to_video_sft.py @@ -536,7 +536,7 @@ def main(args): # We need to initialize the trackers we use, and also store our configuration. # The trackers initializes automatically on the main process. - if accelerator.is_main_process: + if accelerator.distributed_type == DistributedType.DEEPSPEED or accelerator.is_main_process: tracker_name = args.tracker_name or "cogvideox-sft" accelerator.init_trackers(tracker_name, config=vars(args))