__init__.py

from .configuration_afmoe import AfmoeConfig
from .modeling_afmoe     import AfmoeForCausalLM

__all__ = ["AfmoeConfig", "AfmoeForCausalLM"]