a Jb$@sFddlZddlmZddlmZddlmZeeZeddZ dS)N) AutoConfig) AutoModel)register_lang_encodercKsrd}|drt|d}nRt|d}d|vrTtd|d||dtd|t|}|S)NLOAD_PRETRAINEDZHF_MODELZCONFIG_OVERRIDEzOverride config: zHF model config: )rfrom_pretrainedrloggerwarningupdateinfo from_config)config_encoder tokenizerverbosekwargshf_modelZ hf_configr4/datadrive/UniCL-Demo/model/text_encoder/hf_model.py lang_encoder s r) logging transformersrrregistryr getLogger__name__rrrrrrs