File size: 787 Bytes
0b636e9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | """
__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"]
|