include file in commit

Lightning-AI · May 23, 2024 · 4b4f209 · 4b4f209
1 parent e024403
commit 4b4f209
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/extensions/thunder/pretrain.py b/extensions/thunder/pretrain.py
@@ -246,6 +246,7 @@ def fit(
     tokenizer_dir: Optional[Path],
     train: TrainArgs,
     eval: EvalArgs,
+    optimizer: Union[str, Dict],
 ) -> None:
     model = state["model"]
     optimizer = state["optimizer"]