CPM-9G-8B/9G-Train/cpm/cpm9g/__init__.py

6 lines
139 B
Python

from .models import CPM9G
from .models import CPM9GConfig
from .tokenizers import CPM9GTokenizer
from .training_tasks import MixedDataset