storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-09 00:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

wandb tracker in scheduling problems during the training initiation and training stages.

Browse Source
This commit is contained in:
glide-the
2024-11-27 17:14:28 +08:00
parent 62a82a8647
commit a9cf49ecea
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/cogvideox_text_to_video_sft.py
+1 -1
View File
@@ -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))