fix: Fixing issue #82#83

ahsan-z-khan merged 2 commits intoaws:masteraws/sagemaker-inference-toolkit:masterfrom

amaharek:issue-82amaharek/sagemaker-inference-toolkit:issue-82Copy head branch name to clipboard