src/diffusers/utils /outputs.py: Deprecation Warning for torch.utils._pytree._register_pytree_node in latest PyTorch

[zyinghua]

Describe the bug

Hi! The latest version of PyTorch has deprecated torch.utils._pytree._register_pytree_node, recommending the use of torch.utils._pytree.register_pytree_node (without the leading underscore). Perhaps not an urgent issue but wondering might affect long-term compatibility. Happened when I run diffusers/examples/unconditional_image_generation/train_unconditional.py

Reproduction

accelerate launch --multi_gpu train_unconditional.py --train_data_files="your-path-to-data-file/102-flowers.zip" --output_dir="your-path-to-output-folder/ddpm-ema-flowers-64"

Logs

torch.utils._pytree._register_pytree_node(
/root/miniconda3/lib/python3.8/site-packages/diffusers/utils/outputs.py:63: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.

System Info

• diffusers version: 0.27.0.dev0
• Platform: Linux-5.4.0-162-generic-x86_64-with-glibc2.17
• Python version: 3.8.10
• PyTorch version (GPU?): 2.2.0+cu121 (True)
• Huggingface_hub version: 0.20.3
• Transformers version: 4.37.2
• Accelerate version: 0.27.2
• xFormers version: not installed
• Using GPU in script?:
• Using distributed or parallel set-up in script?:

Who can help?

No response

[yiyixuxu]

hi! thanks! do you want to open a PR help us update this? :)

[zyinghua]

hi! thanks! do you want to open a PR help us update this? :)

Hi! Sure happy to do that :)

[floatingwh]

Who know this error ?How to fix it?