transformers
Allow passing `tp_plan` in `from_pretrained` directly
#41435

Merged

Allow passing `tp_plan` in `from_pretrained` directly #41435

Cyrilvallez merged 15 commits into main from tp_plan

start

1a41e7c6

Merge branch 'main' into tp_plan

1bb3fce3

allow passing it

9a24566e

fix plans

4781747c

fix

3d5c40d5

fix

a5abc8d2

style

70af4a45

style

db86c670

Cyrilvallez commented on 2025-10-15

fix

2c86e342

Cyrilvallez commented on 2025-10-15

ArthurZucker approved these changes on 2025-10-15

add_test

ca0b0877

oupsi indent

0653a981

fix

c040a955

fix

d9391b9d

fix for CI without accelerator

75c0cd10

fix import

efbbb7a6

Cyrilvallez merged 3ef6f2c4 into main 161 days ago

Cyrilvallez deleted the tp_plan branch 161 days ago

Reviewers

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone