Fix distributed issue for timm#1653

regisss merged 3 commits into

huggingface:mainhuggingface/optimum-habana:mainfrom

ZhengHongming888:fix_distributed_for_timmZhengHongming888/optimum-habana:fix_distributed_for_timmCopy head branch name to clipboard