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

Merge pull request #100 from a-r-r-o-w/wandb-tracker-fix

Browse Source 
wandb tracker in scheduling problems during the training initiation and training stages
This commit is contained in:
Sayak Paul
2024-11-29 15:08:19 +05:30
committed by GitHub
parent d10963f707 a9cf49ecea
commit d5cc7c66f1
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))