transformers
(Part 2) feat: allow for tp_size attr for tplizing the model
#37054

Merged

(Part 2) feat: allow for tp_size attr for tplizing the model #37054

SunMarc merged 5 commits into huggingface:main from kmehant:tp-size

github-actions marked this pull request as draft 1 year ago

kmehant marked this pull request as ready for review 1 year ago

github-actions requested a review from

Rocketknight1 1 year ago

github-actions requested a review from

stevhliu 1 year ago

SunMarc commented on 2025-03-28

SunMarc requested a review from

ArthurZucker 1 year ago

kmehant force pushed from 72fe5920 to b5b15cdb 1 year ago

kmehant force pushed from a33e9ef6 to b7abb2a3 1 year ago

kmehant force pushed from b7abb2a3 to c7696a2c 1 year ago

SunMarc commented on 2025-04-08

kmehant force pushed from c7696a2c to 71ce444e 1 year ago

kmehant force pushed from 71ce444e to c2d3843c 1 year ago

kmehant force pushed from 307fc4e3 to 33af129d 1 year ago

S1ro1 approved these changes on 2025-04-08

SunMarc approved these changes on 2025-04-08

kmehant force pushed from d06b883a to ccf1889b 1 year ago

kmehant force pushed from ccf1889b to 43bb0717 1 year ago

feat: custom tp_size, new transformers tp interface

1059fffb

fix: review cmt - error when tp_plan not set for tp_size

bb2950d1

fix: nit in docs

a65130c6

kmehant force pushed from 1934f7c6 to a65130c6 1 year ago

Merge branch 'main' into tp-size

d77505c0

Merge branch 'main' into tp-size

c0737368

SunMarc merged 7d768764 into main 1 year ago

ArthurZucker commented on 2025-04-10

Reviewers

SunMarc

S1ro1

ArthurZucker

Rocketknight1

stevhliu

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers (Part 2) feat: allow for tp_size attr for tplizing the model #37054 Merged

(Part 2) feat: allow for tp_size attr for tplizing the model #37054

transformers
(Part 2) feat: allow for tp_size attr for tplizing the model
#37054

Merged