fix

SkafteNicki · SkafteNicki · commit 503c33b85131 · 2025-08-06T09:28:19.000+02:00
diff --git a/src/lightning/pytorch/utilities/model_summary/model_summary.py b/src/lightning/pytorch/utilities/model_summary/model_summary.py
@@ -238,7 +238,7 @@ def __init__(self, model: "pl.LightningModule", max_depth: int = 1) -> None:
             "32-true": 32,
             "64-true": 64,
         }
-        if self._model.trainer is not None and self._model.trainer.precision not in precision_to_bits:
+        if self._model._trainer and self._model.trainer.precision not in precision_to_bits:
             rank_zero_warn(
                 f"Precision {self._model.trainer.precision} is not supported by the model summary. "
                 " Estimated model size in MB will not be accurate. Using 32 bits instead.",

Original file line number	Diff line number	Diff line change
`@@ -238,7 +238,7 @@ def __init__(self, model: "pl.LightningModule", max_depth: int = 1) -> None:`
`238`	`238`	`"32-true": 32,`
`239`	`239`	`"64-true": 64,`
`240`	`240`	`}`
`241`		`- if self._model.trainer is not None and self._model.trainer.precision not in precision_to_bits:`
	`241`	`+ if self._model._trainer and self._model.trainer.precision not in precision_to_bits:`
`242`	`242`	`rank_zero_warn(`
`243`	`243`	`f"Precision {self._model.trainer.precision} is not supported by the model summary. "`
`244`	`244`	`" Estimated model size in MB will not be accurate. Using 32 bits instead.",`