From a9b2202301df9e4d67ffbf8cf6f2d806123e6f7a Mon Sep 17 00:00:00 2001 From: Dipet Date: Fri, 26 Aug 2022 16:40:42 +0300 Subject: [PATCH] Fix OrderedDict import for python3.6 --- deepspeed/runtime/bf16_optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeed/runtime/bf16_optimizer.py b/deepspeed/runtime/bf16_optimizer.py index 303267f0494d..a9988e2c498f 100644 --- a/deepspeed/runtime/bf16_optimizer.py +++ b/deepspeed/runtime/bf16_optimizer.py @@ -2,7 +2,7 @@ Copyright 2022 The Microsoft DeepSpeed Team """ -from typing import OrderedDict +from collections import OrderedDict import torch import os from deepspeed import comm as dist