diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
index 8abe845f1b..688d8deb74 100644
--- a/llmfoundry/callbacks/hf_checkpointer.py
+++ b/llmfoundry/callbacks/hf_checkpointer.py
@@ -195,7 +195,7 @@ def save_model_patch(*args: Any, **kwargs: Any):
                 e,
             )
 
-    mlflow.transformers.save_model = save_model_patch
+    mlflow.transformers.save_model = save_model_patch  # type: ignore
 
     mlflow.set_tracking_uri(mlflow_logger.tracking_uri)
     if mlflow_logger.model_registry_uri is not None: