GatorGPT2 / __init__.py
kunjcr2's picture
Update __init__.py
7e0c21d verified
from .configuration_gator import GatorConfig
from .modeling_gator import GatorModel, GatorModelForCasualLM
__all__ = ["GatorConfig", "GatorModel", "GatorModelForCasualLM"]