ColossalAI/colossalai/utils/model
Jiarui Fang 3ce4463fe6
[utils] remove lazy_memory_allocate from ColoInitContext (#1844)
2022-11-09 11:50:33 +08:00
..
__init__.py fix import error (#880) 2022-04-26 19:28:40 +08:00
colo_init_context.py [utils] remove lazy_memory_allocate from ColoInitContext (#1844) 2022-11-09 11:50:33 +08:00
lazy_init_context.py [hotfix] fix init context (#1543) 2022-09-06 11:45:08 +08:00
utils.py [hotfix] fix init context (#1543) 2022-09-06 11:45:08 +08:00