Lightning-AI / pytorch-lightning UNCLAIMED

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

0 0 0 Python

fix mis-alignment column while using rich model summary in `DeepSpeedstrategy` (#21100)

* fix mis-alignment column while using rich model summary in DeepSpeed strategy.
* test: add minimum gpu requirement in `test_deepspeed_summary_with_rich_model_summary`
* chlog

---------

Co-authored-by: Jirka B <j.borovec+github@gmail.com>
Co-authored-by: Nicki Skafte Detlefsen <skaftenicki@gmail.com>
(cherry picked from commit 2460746464237a787b9d6b34edd3d0c9e061ac86)

GdoongMathew committed 7mo ago

72cc5223e00a01106708cf541cceee32bdf5c72b

Parent: 28c2208

Committed by Luca Antiga <luca@lightning.ai> on 8/29/2025, 10:07:57 AM