transformers
Update parent module attributes when sharding with TP
#44421

Merged

Update parent module attributes when sharding with TP #44421

michaelbenayoun merged 7 commits into huggingface:main from michaelbenayoun:update_layer_attributes_for_tp

michaelbenayoun requested a review from

3outeille 114 days ago

feat: update parent module attribute when sharding weights for TP

81f78deb

style: remove added space

4ccb0017

test: add unit tests for TensorParallelLayer

209047c6

michaelbenayoun force pushed from 8c2efc7e to 209047c6 113 days ago

Merge branch 'main' into update_layer_attributes_for_tp

71ba5c70

Merge branch 'main' into update_layer_attributes_for_tp

6f966475

3outeille approved these changes on 2026-03-05

test: add tests/tensor_parallel/test_tensor_parallel.py to CI

fafb1438

test: add mark to enable test discovery

1e5e6341

michaelbenayoun merged 47092352 into main 112 days ago

michaelbenayoun deleted the update_layer_attributes_for_tp branch 112 days ago

Reviewers

3outeille

Assignees

No one assigned

Labels

None yet

Milestone

No milestone