github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 20:39:59 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
0e616462a7f9e8faaa33d1700a2020ceb03ccd34
ColossalAI/colossalai/checkpoint_io/utils.py
Jun Gao a4489384d5 [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
2023-11-09 17:00:25 +08:00

29 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink