import safetensors.torch

state_dict = safetensors.torch.load_file("/home/kade/source/repos/JTP2/JTP_PILOT2-e3-vit_so400m_patch14_siglip_384.safetensors")
print(state_dict.keys())