.. |
__init__.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
constants.py
|
ce5e56a82e
universal-ckp: support megatron-deepspeed llama model (#4666)
|
11 months ago |
deepspeed_checkpoint.py
|
8998707a2f
Universal Checkpoint for Sequence Parallelism (#4752)
|
10 months ago |
ds_to_universal.py
|
8998707a2f
Universal Checkpoint for Sequence Parallelism (#4752)
|
10 months ago |
reshape_3d_utils.py
|
8998707a2f
Universal Checkpoint for Sequence Parallelism (#4752)
|
10 months ago |
reshape_meg_2d.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
reshape_utils.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
universal_checkpoint.py
|
8998707a2f
Universal Checkpoint for Sequence Parallelism (#4752)
|
10 months ago |
utils.py
|
5c3ebd7ede
Clone tensors to avoid torch.save bloat (#3348)
|
1 year ago |
zero_checkpoint.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |