fix: use main_process_first instead of broadcast_object_list#458

Open

willmj wants to merge 6 commits intofoundation-model-stack:mainfrom willmj:fix-gpu-broadcasting

+4-17

Commits on Feb 5, 2025

fix: use accelerator.main_process_first instead of torch.distributed.broadcast_object_list for multi-GPU
willmj
committed

Commits on Feb 13, 2025

Commits on Feb 14, 2025