Skip to content

weights_only default flip for torch.load #3360

Open
@mikaylagawarecki

Description

@mikaylagawarecki

🐛 Describe the bug

We've flipped the default for the weights_only argument in torch.load to True in pytorch/pytorch, see here for details + documentation and this is coming in torch 2.6.

This is expected to be quite a BC-breaking change, especially if any torch.load calls are not loading plain tensors.

We should make sure that all the torch.load calls in torchserve are still working (see this PR #3073)

This should be merged before updating PyTorch version.

cc @agunapal @mreso

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