From 02c779d1df6e538d883b2817a32030e8a39455b5 Mon Sep 17 00:00:00 2001 From: Yichen Yan Date: Wed, 27 Nov 2024 09:29:22 +0800 Subject: [PATCH] Fix type error in `ZeROOrderedDict` Co-authored-by: Tero Keski-Valkama --- deepspeed/runtime/zero/parameter_offload.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeed/runtime/zero/parameter_offload.py b/deepspeed/runtime/zero/parameter_offload.py index f945f5166190..0be88a1e1ba6 100644 --- 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)