RelaxingSnorlax's picture
Upload folder using huggingface_hub
9d6624a verified
from safetensors.torch import safe_open
file=safe_open("model.safetensors", 'pt')
print(file.keys())
f=file
for k in f.keys():
t = f.get_tensor(k) # torch.Tensor
print(f"{k}: shape={tuple(t.shape)} dtype={t.dtype}")