DeepSpeed
Set tp world size to 1 in ckpt load, if MPU is not provided
#5243

Merged

Set tp world size to 1 in ckpt load, if MPU is not provided #5243

tjruwase merged 1 commit into master from fix_mpu_ckpt

Set tp world size to 1 in ckpt load, if MPU is not provided

b75647c4

samadejacobs requested a review from

tjruwase 1 year ago

samadejacobs requested a review from

mrwyattii 1 year ago

tjruwase approved these changes on 2024-03-08

tjruwase merged 535a908f into master 1 year ago

Reviewers

tjruwase

mrwyattii

Assignees

No one assigned

Labels

None yet

Milestone

No milestone