fix: pass weights_only=True to torch.load#203

anthonyting merged 1 commit intoaws:mainaws/aws-step-functions-data-science-sdk-python:mainfrom

anthonyting:fix-weightsCopy head branch name to clipboard