Bugfix: properly pass kwargs to `torch.jit.load` from `mlfow.pytorch.load_model` (#5163)
Bugfix: properly pass kwargs to `torch.jit.load` from `mlfow.pytorch.load_model` (#5163)

* pass kwargs to `torch.jit.load`

Signed-off-by: Jake Schmidt <1f8f8406a50b7d47647cf92ba91eb02ff39fe376@recursionpharma.com>

* pop pickle_module

Signed-off-by: harupy <3b81b8b110f2244bdbf0f43efa472935c395c613@users.noreply.github.com>

Co-authored-by: harupy <3b81b8b110f2244bdbf0f43efa472935c395c613@users.noreply.github.com>
mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow