github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

storage

pull/520/head
lijiaxing 2023-12-01 14:14:55 +08:00
parent 03e53871a7
commit c7e83fd611
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_utils/common_fixture.py
View File

@ -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")