Making large AI models cheaper, faster and more accessible
[shardformer] Fix serialization error with Tensor Parallel state saving (#5018)
* Fix serialization error with Tensor Parallel state saving * Refactor state_dict CPU transfer using tree_map
J
Jun Gao committed
a4489384d5239074c541820ef583fb5e8371b9ae
Parent: 7244412
Committed by GitHub <noreply@github.com>
on 11/9/2023, 9:00:25 AM