🚀 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