Skip to content

Fix FutureWarning in pretrained.py by setting weights_only=True in torch.load#62

Open
thunn wants to merge 1 commit intogemelo-ai:mainfrom
thunn:main
Open

Fix FutureWarning in pretrained.py by setting weights_only=True in torch.load#62
thunn wants to merge 1 commit intogemelo-ai:mainfrom
thunn:main

Conversation

@thunn
Copy link

@thunn thunn commented Oct 31, 2024

Description:
This PR updates the torch.load call in vocos\pretrained.py to set weights_only=True, following PyTorch's guidance for mitigating potential security risks with pickle deserialization in untrusted environments. (as noted in #61)

  • Tested on PyTorch v2.4.1 with no warning output

@thunn
Copy link
Author

thunn commented Nov 25, 2024

Hi @hubertsiuzdak,

Thank you for your work on this project! This MR addresses a FutureWarning when using vocos. Could you kindly review and merge it if possible? Let me know if any changes are needed.

Appreciate your time!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant