Fix GPT_neox incorrect output with batch query#1358

regisss merged 1 commit into

huggingface:mainhuggingface/optimum-habana:mainfrom

Jianhong-Zhang:gpt_neox_outputJianhong-Zhang/optimum-habana:gpt_neox_outputCopy head branch name to clipboard