Skip to content
2 changes: 1 addition & 1 deletion deepspeed/runtime/zero/parameter_offload.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def __init__(self, parent_module, *args, **kwargs):

def __reduce__(self):
r0, _, *r2 = super().__reduce__()
return (r0, (self._parent_module, )) + r2
return (r0, (self._parent_module, )) + tuple(r2)

def __getitem__(self, key):
param = super().__getitem__(key)
Expand Down