"""
__init__.py — makes hf_hub a proper package and registers
MenteeEmbedConfig + MenteeEmbedModel with the HuggingFace Auto classes.

When HF loads a repo with trust_remote_code=True it imports this file first,
which triggers the AutoModel/AutoConfig registration automatically.
"""
from .configuration_mentee import MenteeEmbedConfig
from .modeling_mentee import MenteeEmbedModel
from .tokenization_mentee import MenteeTokenizer

from transformers import AutoConfig, AutoModel, AutoTokenizer

AutoConfig.register("mentee_embed", MenteeEmbedConfig)
AutoModel.register(MenteeEmbedConfig, MenteeEmbedModel)
AutoTokenizer.register(MenteeEmbedConfig, slow_tokenizer_class=None, fast_tokenizer_class=MenteeTokenizer)

__all__ = ["MenteeEmbedConfig", "MenteeEmbedModel", "MenteeTokenizer"]