mirror of https://github.com/InternLM/InternLM
fix test_checkpoint (#526)
parent
1738bee002
commit
2dbbab7418
|
@ -8,9 +8,8 @@ import torch.distributed as dist
|
|||
from internlm.core.context.parallel_context import Config
|
||||
from internlm.core.trainer import TrainState
|
||||
from internlm.solver.optimizer.hybrid_zero_optim import HybridZeroOptimizer
|
||||
from internlm.utils.common import SingletonMeta
|
||||
from internlm.utils.model_checkpoint import CheckpointManager
|
||||
from internlm.utils.storage_manager import wait_async_upload_finish
|
||||
from internlm.utils.storage_manager import SingletonMeta, wait_async_upload_finish
|
||||
from tests.test_utils.common_fixture import ( # noqa # pylint: disable=unused-import
|
||||
ASYNC_TMP_FOLDER,
|
||||
BOTO_SAVE_PATH,
|
||||
|
|
Loading…
Reference in New Issue