add missing function arg

Lightning-AI · May 23, 2024 · eed1612 · eed1612
1 parent 1521246
commit eed1612
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/litgpt/pretrain.py b/litgpt/pretrain.py
@@ -83,6 +83,7 @@ def setup(
         train: Training-related arguments. See ``litgpt.args.TrainArgs`` for details.
         eval: Evaluation-related arguments. See ``litgpt.args.EvalArgs`` for details.
         optimizer: An optimizer name (such as "AdamW") or config.
+        
         devices: How many devices/GPUs to use. Uses all GPUs by default.
         tokenizer_dir: Optional path to the tokenizer dir that was used for preprocessing the dataset. Only some data
             module require this.