Fix UnboundLocalError in shard_and_distribute_module for replicated parameters#45675

3outeille merged 2 commits intohuggingface:mainhuggingface/transformers:mainfrom

Abdennacer-Badaoui:small-fix-tpAbdennacer-Badaoui/transformers:small-fix-tpCopy head branch name to clipboard