Handle jagged NestedTensor failures in model summary#21780

furk4neg3 wants to merge 2 commits into

Lightning-AI:masterLightning-AI/pytorch-lightning:masterfrom

furk4neg3:fix-jagged-nestedtensor-model-summaryfurk4neg3/pytorch-lightning:fix-jagged-nestedtensor-model-summaryCopy head branch name to clipboard