Add WORLD_SIZE env var for torch.distributed

This commit is contained in:
Connor Hack 2024-11-06 10:16:19 -08:00
parent 307b2ca33e
commit 274b5a0d0c

View file

@ -72,6 +72,7 @@ env:
REPORTS_GEN: ""
RANK: 0
WORLD_SIZE: 4
jobs:
execute_workflow: