From c7e83fd6111879655b2108ff51d248ab3c9b031d Mon Sep 17 00:00:00 2001 From: lijiaxing Date: Fri, 1 Dec 2023 14:14:55 +0800 Subject: [PATCH] storage --- tests/test_utils/common_fixture.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_utils/common_fixture.py b/tests/test_utils/common_fixture.py index 746f43e..e5d6328 100644 --- a/tests/test_utils/common_fixture.py +++ b/tests/test_utils/common_fixture.py @@ -8,6 +8,7 @@ import torch from internlm.core.context import global_context as gpc from internlm.core.context.parallel_context import Config from internlm.solver.optimizer.hybrid_zero_optim import HybridZeroOptimizer +from internlm.train.utils import create_param_groups from internlm.utils.storage_manager import SingletonMeta OSS_NAME = os.environ.get("OSS_BUCKET_NAME")