refactor the code and add wrap_in_hpu_graph to corner case#625

regisss merged 3 commits intohuggingface:mainhuggingface/text-embeddings-inference:mainfrom

kaixuanliu:corner-casekaixuanliu/text-embeddings-inference:corner-caseCopy head branch name to clipboard