Formatting.

mlfoundations · Dec 11, 2023 · 51f889b · 51f889b
1 parent 2dcf57b
commit 51f889b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/open_lm/main.py b/open_lm/main.py
@@ -243,7 +243,9 @@ def main(args):
     # fully initialize distributed device environment
     device = init_distributed_device(args)
 
-    assert args.global_batch_size % args.world_size == 0, "Global batch size is not divisible by number of GPUs, and thus cannot be respected."
+    assert (
+        args.global_batch_size % args.world_size == 0
+    ), "Global batch size is not divisible by number of GPUs, and thus cannot be respected."
 
     args.per_gpu_batch_size = args.global_batch_size // args.world_size