Set weights_only in torch.load calls

Set weights_only in torch.load calls #54