| fix(pytorch):add ckpt-format argument to scripts
Co-authored-by: z__y<z4t155664@163.com>
# message auto-generated for no-merge-commit merge:
!4371 merge add_ckpt_torch_dist_argument_for_shells into master
fix(pytorch):add ckpt-format argument to scripts
Created-by: z__y
Commit-by: z__y
Merged-by: ascend-robot
Description: ## What this PR does / why we need it?
This PR explicitly adds ckpt-format torch to all repository scripts to support the asynchronous checkpoint saving feature.
## Does this PR introduce any user-facing change?
No. This change only adjusts internal script parameters to maintain existing behavior. There are no user-facing API or usage changes.
## How was this patch tested?
Tests confirm that asynchronous checkpoint saving works correctly and that the original torch format checkpoint behavior is preserved.
See merge request: Ascend/MindSpeed-LLM!4371 | 1 个月前 |
