Description
🚀 The feature, motivation and pitch
pytorch.load with weights_only=True allows for loading of lists & dicts
I would expect the builtin set to be just as safe as a dict
Alternatives
Another alternative would be to convert sets to lists before saving, then converting back to a set when loading if the model needs a set internally
Additional context
WeightsUnpickler error: Unsupported global: GLOBAL builtins.set was not an allowed global by default. Please use `torch.serialization.add_safe_globals([set])` to allowlist this global if you trust this class/function.
cc @mruberry @mikaylagawarecki @seemethere @malfet @pytorch/pytorch-dev-infra
Metadata
Metadata
Assignees
Labels
Type
Projects
Status