Skip to content

ml_connect_load() issue with PyTorch 2.6+ #132

@edgararuiz

Description

@edgararuiz
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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions