Skip to content

Commit 17654cb

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 559f15a commit 17654cb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_zero/test_gemini/test_inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from colossalai.accelerator import get_accelerator
1111
from colossalai.legacy.amp import convert_to_apex_amp
1212
from colossalai.nn.optimizer import HybridAdam
13-
from colossalai.testing import DummyDataloader, parameterize, rerun_if_address_is_in_use, spawn, clear_cache_before_run
13+
from colossalai.testing import DummyDataloader, clear_cache_before_run, parameterize, rerun_if_address_is_in_use, spawn
1414
from colossalai.utils import set_seed
1515
from colossalai.zero import GeminiDDP, GeminiOptimizer
1616
from colossalai.zero.gemini.chunk import search_chunk_configuration

0 commit comments

Comments
 (0)