and get access to the augmented documentation experience

pipinstall safetensors

condainstall -c huggingface safetensors

from safetensorsimport safe_opentensors = {}with safe_open("model.safetensors", framework="pt", device=0)as f:for kin f.keys():        tensors[k] = f.get_tensor(k)

from safetensorsimport safe_opentensors = {}with safe_open("model.safetensors", framework="pt", device=0)as f:    tensor_slice = f.get_slice("embedding")    vocab_size, hidden_dim = tensor_slice.get_shape()    tensor = tensor_slice[:, :hidden_dim]

import torchfrom safetensors.torchimport save_filetensors = {"embedding": torch.zeros((2,2)),"attention": torch.zeros((2,3))}save_file(tensors,"model.safetensors")