mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Adds the following to allowed globals for the `weights_only` unpickler - [x] `torch._utils._rebuild_qtensor` and qtensor related types - [x] `torch._utils._rebuild_parameter_with_state` (used deserializing a parameter that has user-defined attributes like `Param.foo`) The remaining rebuild functions that have not been allowlisted are - [x] `torch._utils._rebuild_wrapper_subclass` (allowlisted in above PR) - [ ] `torch._utils._rebuild_device_tensor_from_numpy` - [ ] `torch._utils._rebuild_xla_tensor` (legacy) Pull Request resolved: https://github.com/pytorch/pytorch/pull/124330 Approved by: https://github.com/albanD |
||
|---|---|---|
| .. | ||
| ao_migration | ||
| bc | ||
| core | ||
| eager | ||
| fx | ||
| jit | ||
| pt2e | ||
| serialized | ||
| __init__.py | ||