Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
check param is of `nn.Parameter` type for pruning sanitization (#20783)
* check param is of nn.Parameter type * Update test_pruning.py --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Rittik Panda <[email protected]>
G
Garibaldi Pineda-Garcia committed
cb1afbe37772c2b3acbbeb062703c0020355de13
Parent: 76d3d22
Committed by GitHub <[email protected]>
on 6/18/2025, 11:05:02 PM