transformers
47092352 - Update parent module attributes when sharding with TP (#44421)

Commit
2 days ago
Update parent module attributes when sharding with TP (#44421) * feat: update parent module attribute when sharding weights for TP * style: remove added space * test: add unit tests for TensorParallelLayer * test: add tests/tensor_parallel/test_tensor_parallel.py to CI * test: add mark to enable test discovery --------- Co-authored-by: Ferdinand Mom <47445085+3outeille@users.noreply.github.com>
Parents
Loading