def test_safetensors_load_from_hub_from_safetensors_pt(self):
        """
        This test checks that we can load safetensors from a checkpoint that only has those on the Hub.
        saved in the "pt" format.
        """
        tf_model = TFBertModel.from_pretrained("hf-internal-testing/tiny-bert-h5")

        # Can load from the PyTorch-formatted checkpoint
        safetensors_model = TFBertModel.from_pretrained("hf-internal-testing/tiny-bert-pt-safetensors")
        for p1, p2 in zip(tf_model.weights, safetensors_model.weights):
            self.assertTrue(np.allclose(p1.numpy(), p2.numpy()))

        This test checks that we can load safetensors from a checkpoint that only has those on the Hub.
        saved in the "pt" format.
        
dpkd/transformers/tests/test_modeling_tf_utils.py