pyright

mosaicml · Sep 27, 2023 · 62a0fd6 · 62a0fd6
1 parent 6180466
commit 62a0fd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -195,7 +195,7 @@ def _save_checkpoint(self, state: State, logger: Logger):
                             overwrite=self.overwrite,
                         )
 
-                if self.log_to_mlflow and state.get_elapsed_duration() >= 1.0:
+                if self.log_to_mlflow and state.get_elapsed_duration() is not None and state.get_elapsed_duration() >= 1.0:
                     log.debug('Reloading model to log to MLFlow')
 
                     from torch.distributed.fsdp import \