Skip to content

GitLab

Commit 36ab2cb7 authored by oahzxl's avatar oahzxl
Browse files

change import

parent 7ab2db20
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
tests/test_autochunk/test_autochunk_codegen.py
View file @ 36ab2cb7
...@@ -12,7 +12,6 @@ from colossalai.fx._compatibility import is_compatible_with_meta ...@@ -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.codegen.activation_checkpoint_codegen import CODEGEN_AVAILABLE
from colossalai.fx.graph_module import ColoGraphModule from colossalai.fx.graph_module import ColoGraphModule
from colossalai.fx.passes.meta_info_prop import MetaInfoProp from colossalai.fx.passes.meta_info_prop import MetaInfoProp
from colossalai.fx.profiler import MetaTensor
from colossalai.utils import free_port from colossalai.utils import free_port
from tests.test_autochunk.evoformer.evoformer import evoformer_base from tests.test_autochunk.evoformer.evoformer import evoformer_base
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment