proxy_bert.py

try:
    from mblt_model_zoo.hf_transformers.models.bert.configuration_bert import (
        MobilintBertConfig,
    )
    from mblt_model_zoo.hf_transformers.models.bert.modeling_bert import (
        MobilintBertForMaskedLM,
    )
except ImportError:
    raise ImportError(
        "This model requires 'mblt_model_zoo' to be installed. "
        "Please run: pip install mblt_model_zoo[transformers]"
    )

__all__ = ["MobilintBertConfig", "MobilintBertForMaskedLM"]