pytorch-lightning

345450b0 - Fix parameter count in ModelSummary when parameters are DTensors (#20163)