From 30efbaea77dd600e9d8e7e8e411fd9b58ce5aa43 Mon Sep 17 00:00:00 2001 From: fw Date: Tue, 3 Sep 2024 08:28:14 +0000 Subject: [PATCH] . --- realhf/base/topology.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/realhf/base/topology.py b/realhf/base/topology.py index fd77fd2d..c24a10d1 100755 --- a/realhf/base/topology.py +++ b/realhf/base/topology.py @@ -1,6 +1,7 @@ # Modified from https://github.com/microsoft/DeepSpeed/blob/aed599b4422b1cdf7397abb05a58c3726523a333/deepspeed/runtime/pipe/topology.py# -from itertools import product as cartesian_product, permutations +from itertools import permutations +from itertools import product as cartesian_product from typing import Dict, List, NamedTuple, Optional, Tuple import torch.distributed as dist