code: def test_efficientnet_safetensors(self):
      from extra.models.efficientnet import EfficientNet
  
      model = EfficientNet(0)
      state_dict = get_state_dict(model)
      safe_save(state_dict, temp("eff0"))
      state_dict_loaded = safe_load(temp("eff0"))
      assert sorted(state_dict_loaded.keys()) == sorted(
          state_dict.keys()
      )
      for k, v in state_dict.items():
          np.testing.assert_array_equal(
              v.numpy(), state_dict_loaded[k].numpy()
          )
  
      # load with the real safetensors
      from safetensors import safe_open
  
      with safe_open(temp("eff0"), framework="pt", device="cpu") as f:
          assert sorted(f.keys()) == sorted(state_dict.keys())
          for k in f.keys():
              np.testing.assert_array_equal(
                  f.get_tensor(k).numpy(), state_dict[k].numpy()
              )
  
imports: import pathlib, unittest
  import numpy
  from tinygrad import Tensor, Device, dtypes
  from tinygrad.nn.state import safe_load, safe_save, get_state_dict, torch_load
  from tinygrad.helpers import Timing, CI, fetch, temp
label_desc: Use image node_id 3 for calling the TestSafetensors obj's underlying member method code with example usage: obj.test_efficientnet_safetensors() without return types