PR #4854 add sharded loading for safetensors in AutoTP

add sharded loading for safetensors in AutoTP #4854

mrwyattii merged 3 commits into deepspeedai:master from sywangyi:safetensor_autoTP

sywangyi requested a review from

mrwyattii 2 years ago

sywangyi requested a review from

awan-10 2 years ago

sywangyi requested a review from

cmikeh2 2 years ago

sywangyi requested a review from

arashb 2 years ago

sywangyi requested a review from

loadams 2 years ago

mrwyattii approved these changes on 2023-12-21

sywangyi force pushed from e156a245 to 18fa621d 2 years ago

add sharded loading for safetensors in AutoTP

f7e8d572

sywangyi force pushed from 50cb9374 to f7e8d572 2 years ago

Merge branch 'master' into safetensor_autoTP

d747e75e

loadams approved these changes on 2024-01-02

Merge branch 'master' into safetensor_autoTP

35d90b36

mrwyattii merged c8c57b8c into master 2 years ago

sywangyi deleted the safetensor_autoTP branch 2 years ago

Reviewers

loadams

mrwyattii

awan-10

cmikeh2

arashb

Assignees

No one assigned

Labels

None yet

Milestone

No milestone