mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-01 11:55:23 +00:00
8 lines
197 B
Python
8 lines
197 B
Python
#!/usr/bin/env python
|
|
# -*- encoding: utf-8 -*-
|
|
|
|
from colossalai.context import ParallelContext, MoeContext
|
|
|
|
global_context = ParallelContext.get_instance()
|
|
moe_context = MoeContext.get_instance()
|