mirror of https://github.com/hpcaitech/ColossalAI
4 lines
99 B
Python
4 lines
99 B
Python
|
from .embed import vocab_parallel_cross_entropy
|
||
|
|
from .gpt1d import *
|
||
|
|
from .pipeline_gpt1d import *
|