docstr: This test checks that we can load safetensors from a checkpoint that only has those on the Hub
docstr_tok: ["This","test","checks","that","we","can","load","safetensors","from","a","checkpoint","that","only","has","those","on","the","Hub"]
code: def test_safetensors_load_from_local(self):
      """
      This test checks that we can load safetensors from a checkpoint that only has those on the Hub
      """
      with tempfile.TemporaryDirectory() as tmp:
          location = snapshot_download(
              "hf-internal-testing/tiny-bert-flax-only", cache_dir=tmp
          )
          flax_model = FlaxBertModel.from_pretrained(location)
  
      with tempfile.TemporaryDirectory() as tmp:
          location = snapshot_download(
              "hf-internal-testing/tiny-bert-flax-safetensors-only",
              cache_dir=tmp,
          )
          safetensors_model = FlaxBertModel.from_pretrained(location)
  
      self.assertTrue(check_models_equal(flax_model, safetensors_model))
  
imports: import tempfile
  import unittest
  import numpy
  from huggingface_hub import HfFolder, delete_repo, snapshot_download
  from requests.exceptions import HTTPError
  from transformers import BertConfig, BertModel, is_flax_available, is_torch_available
  from transformers.testing_utils import TOKEN, USER, is_pt_flax_cross_test, is_staging_test, require_flax, require_safetensors, require_torch
  from transformers.utils import FLAX_WEIGHTS_NAME, SAFE_WEIGHTS_NAME
label_desc: Use image node_id 8 for calling the FlaxModelUtilsTest obj's underlying member method code with example usage: obj.test_safetensors_load_from_local() without return types