from RetNet.retnet.modeling_retnet import RetNetForCausalLM, RetNetConfig


LightNovelIntroRetNetModelConfig = RetNetConfig


class LightNovelIntroRetNetModel(RetNetForCausalLM):
    config_class = LightNovelIntroRetNetModelConfig