Fix for Falcon-40b inference with deepspeed#502

regisss merged 2 commits into

mainhuggingface/optimum-habana:mainfrom

dev/schoi/falcon_dshuggingface/optimum-habana:dev/schoi/falcon_dsCopy head branch name to clipboard