transformers
TP initialization module-by-module
#35996
Merged

TP initialization module-by-module #35996

ArthurZucker merged 13 commits into main from tp-loading
Cyrilvallez
Cyrilvallez module-by-module loading!
5e198a0e
Cyrilvallez Update modeling_utils.py
190fd7dd
Cyrilvallez dtyle and comments
b7aa37c6
Cyrilvallez Update modeling_utils.py
321f8ee7
Cyrilvallez Update modeling_utils.py
51f0aa02
Cyrilvallez Update test
2f7e20c0
Cyrilvallez Update modeling_utils.py
6ca18389
Cyrilvallez Update modeling_utils.py
8c419c6f
HuggingFaceDocBuilderDev
Cyrilvallez Update test_tp.py
a3a55d06
Cyrilvallez Update test_tp.py
9ab85397
Cyrilvallez Update modeling_utils.py
0d8e55a1
Cyrilvallez re-trigger CIs
cb5d92b3
Cyrilvallez re-trigger CIs
636a3889
ArthurZucker
ArthurZucker commented on 2025-02-03
ArthurZucker
ArthurZucker approved these changes on 2025-02-19
ArthurZucker ArthurZucker merged 60226c6f into main 1 year ago
ArthurZucker ArthurZucker deleted the tp-loading branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone