ColossalAI/examples/language/gpt/titans/model/__init__.py

4 lines
99 B
Python

from .embed import vocab_parallel_cross_entropy
from .gpt1d import *
from .pipeline_gpt1d import *