Skip to content

Commit bcc8c66

Browse files
committed
update model message
1 parent 1af7f73 commit bcc8c66

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

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

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,9 +106,8 @@ def _configure_checkpoint_callbacks(self, enable_checkpointing: bool) -> None:
106106
model_checkpoint = LitModelCheckpoint(model_registry=self.trainer._model_registry)
107107
else:
108108
rank_zero_info(
109-
"You are using the default ModelCheckpoint callback."
110-
" Install `pip install litmodels` package to use the `LitModelCheckpoint` instead"
111-
" for seamless uploading to the Lightning model registry."
109+
"Using default `ModelCheckpoint`. Consider installing `litmodels` package to enable"
110+
" `LitModelCheckpoint` for automatic upload to the Lightning model registry."
112111
)
113112
model_checkpoint = ModelCheckpoint()
114113
self.trainer.callbacks.append(model_checkpoint)

0 commit comments

Comments
 (0)