lightning_pytree
Module for patching PyTorch Lightning's internal pytree utility to avoid FutureWarning related to LeafSpec construction in newer PyTorch versions.
apply_lightning_pytree_leafspec_patch()
Idempotent runtime patch; no-op if torch / Lightning are missing or torch is too old.
| Parameter | Type | Default | Description |
|---|---|---|---|
Returns: None.