| 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}") | |
| 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}") | |