U sNb$@sFddlZddlmZddlmZddlmZeeZeddZ dS)N) AutoConfig) AutoModel)register_lang_encodercKsrd}|drt|d}nRt|d}d|krTtd|d||dtd|t|}|S)NLOAD_PRETRAINEDZHF_MODELZCONFIG_OVERRIDEzOverride config: zHF model config: )rfrom_pretrainedrloggerwarningupdateinfo from_config)config_encoder tokenizerverbosekwargshf_modelZ hf_configr//datadrive/UniCL/model/text_encoder/hf_model.py lang_encoder s r) logging transformersrrregistryr getLogger__name__rrrrrrs