update model message

.github/workflows/docs-build.yml

@@ -72,7 +72,7 @@ jobs:
           lfs: ${{ matrix.pkg-name == 'pytorch' }}
       - uses: actions/setup-python@v5
         with:
-          python-version: "3.9"
+          python-version: "3.10"
 
       - name: List notebooks
         if: ${{ matrix.pkg-name == 'pytorch' }}

docs/source-fabric/conf.py

@@ -292,6 +292,8 @@
     ("py:.*", "torch_xla.*"),
     ("py:class", "transformer_engine.*"),
     ("py:class", "bitsandbytes.*"),
+    # loggers
+    ('py:class', 'tensorboardX.SummaryWriter'),  # todo: this is unexpected as the imports locally works
 ]
 
 # -- Options for todo extension ----------------------------------------------

docs/source-pytorch/conf.py

@@ -377,6 +377,7 @@ def _load_py_module(name: str, location: str) -> ModuleType:
     # missing in generated API
     ("py:exc", "MisconfigurationException"),
     # TODO: generated list of all existing ATM, need to be fixed
+    ('py:class', 'tensorboardX.SummaryWriter'),
     ("py:class", "AveragedModel"),
     ("py:class", "CometExperiment"),
     ("py:meth", "DataModule.__init__"),

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

@@ -106,9 +106,8 @@ def _configure_checkpoint_callbacks(self, enable_checkpointing: bool) -> None:
                 model_checkpoint = LitModelCheckpoint(model_registry=self.trainer._model_registry)
             else:
                 rank_zero_info(
-                    "You are using the default ModelCheckpoint callback."
-                    " Install `pip install litmodels` package to use the `LitModelCheckpoint` instead"
-                    " for seamless uploading to the Lightning model registry."
+                    "Using default `ModelCheckpoint`. Consider installing `litmodels` package to enable"
+                    " `LitModelCheckpoint` for automatic upload to the Lightning model registry."
                 )
                 model_checkpoint = ModelCheckpoint()
             self.trainer.callbacks.append(model_checkpoint)