Skip to content

Commit b4794f7

Browse files
committed
update
1 parent 68499b9 commit b4794f7

1 file changed

Lines changed: 2 additions & 5 deletions

File tree

src/lightning/pytorch/trainer/connectors/callback_connector.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -108,11 +108,8 @@ def _configure_checkpoint_callbacks(self, enable_checkpointing: bool) -> None:
108108

109109
model_checkpoint = LitModelCheckpoint(model_registry=self.trainer._model_registry)
110110
else:
111-
rank_zero_info(
112-
"💡 Tip: For seamless cloud uploads and versioning,"
113-
" try installing [litmodels](https://pypi.org/project/litmodels/) to enable LitModelCheckpoint,"
114-
" which syncs automatically with the Lightning model registry."
115-
)
111+
# Defer the litmodels tip until loggers are set up (in _attach_model_callbacks)
112+
self._pending_litmodels_tip = True
116113
model_checkpoint = ModelCheckpoint()
117114
self.trainer.callbacks.append(model_checkpoint)
118115

0 commit comments

Comments
 (0)