(Part 2) feat: allow for tp_size attr for tplizing the model#37054

SunMarc merged 5 commits intohuggingface:mainhuggingface/transformers:mainfrom

kmehant:tp-sizekmehant/transformers:tp-sizeCopy head branch name to clipboard