U
    T?h#                  7   @   s  d ddddgZ dddgdd	d
fdddgdd	d
fdddgdd	d
fdgdd	dfdgdd	dfdgdd	dfdgdddfdgdddfdgdd	dfddgdd	d
fdgdd	d
fdgdd	d
fdgddd
fdgdd	d
fdgdd	d
fddgdd	d
fddgdd	d
fddgdd	d
fddgdd	d
fddgdd	d
fddgdd	d
fddgdd	d
fdgddd
fdgdd	d
fdgdd	d
fdgdd	d
fdgddd
fdgdd	d
fdgdd	d
fdgddd
fdgdd	d
fdgddd
fdgdd	d
fdgdd	d
fddgdd	dfddgdd	dfddgdd	dfddgdd	dfdgdd	dfdgdd	dfdgddd
fdgddd
fdgdd	d
fdgdd	d
fdgdd	d
fdgdd	d
fdgdd	d
fdddgdd	d
fdgdd	dfdgdd	dfdgdd	dfdgdd	dfd4ZdS )Z	AutoModelZAutoModelWithLMHeadZ"AutoModelForSequenceClassificationZAutoModelForQuestionAnsweringZAutoModelForCausalLMZ	input_idsZattention_maskZtoken_type_ids   FZbert   gpt2TZmemsZbartZvisual_featsZ
visual_posZpixel_valuesZvitZswin)4zbert-base-uncasedzbert-large-uncasedzbert-base-casedz
openai-gptr   zgpt2-mediumz
gpt2-largezgpt2-xlZ
distilgpt2ztransfo-xl-wt103zxlnet-base-casedzxlnet-large-casedzxlm-mlm-en-2048zxlm-mlm-ende-1024zxlm-mlm-enfr-1024zroberta-basezroberta-largezroberta-large-mnlizdeepset/roberta-base-squad2zdistilroberta-basezdistilbert-base-uncasedz'distilbert-base-uncased-distilled-squadZctrlzcamembert-basezalbert-base-v1zalbert-large-v1zalbert-xlarge-v1zalbert-base-v2zalbert-large-v2zalbert-xlarge-v2zxlm-roberta-basezxlm-roberta-largezflaubert/flaubert_small_casedzflaubert/flaubert_base_casedzfacebook/bart-largezfacebook/bart-basezfacebook/bart-large-mnlizfacebook/bart-large-cnnzmicrosoft/DialoGPT-smallzmicrosoft/DialoGPT-mediumzfacebook/mbart-large-cc25zfacebook/mbart-large-en-rozmicrosoft/layoutlm-base-uncasedz microsoft/layoutlm-large-uncasedzsqueezebert/squeezebert-uncasedzsqueezebert/squeezebert-mnliz%squeezebert/squeezebert-mnli-headlesszunc-nlp/lxmert-base-uncasedzgoogle/vit-base-patch16-224z&microsoft/swin-base-patch4-window7-224z'microsoft/swin-small-patch4-window7-224z&microsoft/swin-tiny-patch4-window7-224N)ZMODEL_CLASSESZMODELS r   r   ]/var/www/html/venv/lib/python3.8/site-packages/onnxruntime/transformers/huggingface_models.py<module>	   s   
	 