Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ml_connect_load() issue with PyTorch 2.6+ #132

Open
edgararuiz opened this issue Feb 11, 2025 · 0 comments
Open

ml_connect_load() issue with PyTorch 2.6+ #132

edgararuiz opened this issue Feb 11, 2025 · 0 comments

Comments

@edgararuiz
Copy link
Collaborator

ml_connect_load(sc, fitted_folder)

Error in py_call_impl(callable, call_args$unnamed, call_args$named) : 
  _pickle.UnpicklingError: Weights only load failed. This file can still be loaded, to do so you have two options, do those 
steps only if you trust the source of the checkpoint. 
	(1) In PyTorch 2.6, we changed the default value of the `weights_only` argument in `torch.load` from `False` to 
`True`. Re-running `torch.load` with `weights_only` set to `False` will likely succeed, but it can result in arbitrary 
code execution. Do it only if you got the file from a trusted source.
<...truncated...>by default. Please use `torch.serialization.add_safe_globals([Sequential])` or the 
`torch.serialization.safe_globals([Sequential])` context manager to allowlist this global if you trust this
 class/function.

Check the documentation of torch.load to learn more about types accepted by default with
 weights_only https://pytorch.org/docs/stable/generated/torch.load.html.
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

No branches or pull requests

1 participant