Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[shardformer] Fix serialization error with Tensor Parallel state savi…
…ng (#5018) * Fix serialization error with Tensor Parallel state saving * Refactor state_dict CPU transfer using tree_map
- Loading branch information