log predict time properly#27

philschmid merged 2 commits intoaws:mainaws/sagemaker-huggingface-inference-toolkit:mainfrom

dzorlu:maindzorlu/sagemaker-huggingface-inference-toolkit:mainCopy head branch name to clipboard