.. |
__init__.py
|
2c51aba0b7
Add custom reshaping for universal checkpoint (#5390)
|
6 月之前 |
constants.py
|
2c51aba0b7
Add custom reshaping for universal checkpoint (#5390)
|
6 月之前 |
deepspeed_checkpoint.py
|
2fc702ed9f
DeepSpeedCheckpoint: support custom final ln idx (#5506)
|
4 月之前 |
ds_to_universal.py
|
f0e3f01d7c
Add an argument to enable the injection of missing state during the conversion of universal checkpoints (#5608)
|
3 月之前 |
reshape_3d_utils.py
|
8998707a2f
Universal Checkpoint for Sequence Parallelism (#4752)
|
10 月之前 |
reshape_meg_2d.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
reshape_utils.py
|
c946a34220
Fix sort of zero checkpoint files (#5342)
|
6 月之前 |
universal_checkpoint.py
|
2c51aba0b7
Add custom reshaping for universal checkpoint (#5390)
|
6 月之前 |
utils.py
|
5c3ebd7ede
Clone tensors to avoid torch.save bloat (#3348)
|
1 年之前 |
zero_checkpoint.py
|
c56a4b9e0d
Improve universal checkpoint (#5289)
|
6 月之前 |