transformers
Add default TP plan for all models with backend support
#35870
Merged

Add default TP plan for all models with backend support #35870

ArthurZucker merged 14 commits into main from add-tp
Cyrilvallez
Cyrilvallez Cyrilvallez changed the title Add default TP plan (for all models with backend support) Add default TP plan for all models with backend support 1 year ago
HuggingFaceDocBuilderDev
ArthurZucker
ArthurZucker approved these changes on 2025-01-27
ArthurZucker
Cyrilvallez Add some tp plans!
3b39dda5
Cyrilvallez More tp plans!
3eae346a
Cyrilvallez Add it in the comment
b2a33cb1
Cyrilvallez style
2e53552d
Cyrilvallez Update configuration_mixtral.py
d2fd26a8
Cyrilvallez Update configuration_phi.py
6114bb37
Cyrilvallez update the layout according to special archs
83d208b3
Cyrilvallez fix mixtral
c7eb39eb
Cyrilvallez style
49659d94
Cyrilvallez trigger CIs
da5fe0f1
Cyrilvallez trigger CIs
021ccec6
Cyrilvallez Cyrilvallez force pushed from 9630afbe to 021ccec6 1 year ago
Cyrilvallez CIs
fcd8193a
Cyrilvallez olmo2
165c400c
ArthurZucker Merge branch 'main' into add-tp
c9347669
ArthurZucker ArthurZucker merged 3613f568 into main 1 year ago
ArthurZucker ArthurZucker deleted the add-tp branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone