Fix type error in ZeROOrderedDict

microsoft · Nov 27, 2024 · 5351b08 · 5351b08
1 parent f743fec
commit 5351b08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/runtime/zero/parameter_offload.py b/deepspeed/runtime/zero/parameter_offload.py
@@ -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)