Set default

mosaicml · Feb 5, 2024 · a957282 · a957282
1 parent c343660
commit a957282
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -34,7 +34,8 @@
 
 
 def _maybe_get_license_filename(
-        local_dir: str, pretrained_model_name: Optional[str]) -> Optional[str]:
+        local_dir: str,
+        pretrained_model_name: Optional[str] = None) -> Optional[str]:
     """Returns the name of the license file if it exists in the local_dir.
 
     Note: This is intended to be consistent with the code in MLflow.