unet/open.py

from safetensors import safe_open

tensors = {}
with safe_open("diffusion_pytorch_model.safetensors", framework="pt", device=0) as f:
    for k in f.keys():
        print(k)