__init__.py

from transformers import AutoConfig, AutoProcessor, AutoModel

from .configuration_rynn_value_lang import RynnValueLangConfig
from .conversations import (
    ConversationBuilder,
    InterleavedHistoryConversationBuilder,
    build_conversation_builder,
)
from .modeling_rynn_value_lang import RynnValueLangModel
from .processing_rynn_value_lang import RynnValueLangProcessor

from . import attention_impl


__all__ = [
    "RynnValueLangConfig",
    "RynnValueLangModel",
    "RynnValueLangProcessor",
    "ConversationBuilder",
    "InterleavedHistoryConversationBuilder",
    "build_conversation_builder",
    "attention_impl"
]


AutoConfig.register("rynn_value_lang", RynnValueLangConfig)
AutoModel.register(RynnValueLangConfig, RynnValueLangModel)
AutoProcessor.register(RynnValueLangConfig, RynnValueLangProcessor)