Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
Remove redundant code to set the device on the LightningModule (#19877)
Co-authored-by: Luca Antiga <[email protected]>
A
awaelchli committed
82e6e61bea85c27d869b594f7791383c7a4efaf3
Parent: d5bf4b9
Committed by GitHub <[email protected]>
on 5/20/2024, 4:29:37 AM