convert_diffusers_to_original_sdxl.py doesn't work anymore

[zyddnys]

Describe the bug

Nowadays models are saved as diffusion_pytorch_model-00001-of-00002.safetensors and diffusion_pytorch_model-00002-of-00002.safetensors instead of a single diffusion_pytorch_model.safetensors
This causes convert_diffusers_to_original_sdxl.py to generate file not found error

FileNotFoundError: [Errno 2] No such file or directory: 'output/XXX/unet/diffusion_pytorch_model.bin'

Reproduction

python convert_diffusers_to_original_sdxl.py --model_path output/XXX/ --checkpoint XXX.safetensors --use_safetensors

Logs

No response

System Info

diffusers=0.29.2

Who can help?

No response

[Dhruv Nair]

cc: @sayakpaul

[piaoliang li]

meet this issue too

[piaoliang li]

I solve this issue by adding these line :

Load models from safetensors if it exists, if it doesn't pytorch

if osp.exists(unet_path):
    unet_state_dict = load_file(unet_path, device="cpu")
###  modify:
elif osp.exists(osp.join(args.model_path, "unet", "diffusion_pytorch_model.safetensors.index.json")):
    dict_list = glob.glob((args.model_path + "/" +"unet" + "/*.safetensors"))
    unet_state_dict = {}
    for dict_path in dict_list:
        part_dict = {}
        with safe_open(dict_path, framework="pt", device="cpu") as f:
            for k in f.keys():
                part_dict[k] = f.get_tensor(k)
        unet_state_dict.update(part_dict)
else:
    unet_path = osp.join(args.model_path, "unet", "diffusion_pytorch_model.bin")
    unet_state_dict = torch.load(unet_path, map_location="cpu")

[Sayak Paul]

Thanks for reporting. Would you be able to open a PR for this now that you have a solution too? Please also provide a full reproducible example that we can verify on our end to confirm that the fix is working.

[Lei]

meet this issue too

[John Slegers]

Related to #8974