mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-24 14:33:28 +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()
|