ColossalAI/colossalai/legacy/context
Hongxin Liu 1f5d2e8062
[hotfix] fix torch 2.0 compatibility (#4936)
* [hotfix] fix launch

* [test] fix test gemini optim

* [shardformer] fix vit
2023-10-18 11:05:25 +08:00
..
process_group_initializer [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
random [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
__init__.py [legacy] clean up legacy code (#4743) 2023-09-18 16:31:06 +08:00
parallel_context.py [hotfix] fix torch 2.0 compatibility (#4936) 2023-10-18 11:05:25 +08:00
parallel_mode.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00