[WebDataset] Add `.pth` support for torch tensors

In this PR I add support for `.pth` but with `weights_only=True` to disallow the use of pickle