From 36ab2cb783fdee9899f734f99633fc0d63d4e980 Mon Sep 17 00:00:00 2001 From: oahzxl Date: Tue, 10 Jan 2023 12:20:40 +0800 Subject: [PATCH] change import --- tests/test_autochunk/test_autochunk_codegen.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/test_autochunk/test_autochunk_codegen.py b/tests/test_autochunk/test_autochunk_codegen.py index fe1916884..a061e0ce1 100644 --- a/tests/test_autochunk/test_autochunk_codegen.py +++ b/tests/test_autochunk/test_autochunk_codegen.py @@ -12,7 +12,6 @@ from colossalai.fx._compatibility import is_compatible_with_meta from colossalai.fx.codegen.activation_checkpoint_codegen import CODEGEN_AVAILABLE from colossalai.fx.graph_module import ColoGraphModule from colossalai.fx.passes.meta_info_prop import MetaInfoProp -from colossalai.fx.profiler import MetaTensor from colossalai.utils import free_port from tests.test_autochunk.evoformer.evoformer import evoformer_base