Lower complexity of get_balanced_memory by adding a set#3776

SunMarc merged 3 commits intohuggingface:mainhuggingface/accelerate:mainfrom

SamuelBarryCS:optimize-get-balance-memorySamuelBarryCS/accelerate:optimize-get-balance-memoryCopy head branch name to clipboard