U
    &c                     @   s   d d Z edkre   dS )c                  C   s  dd l } t| jdk s2t| jdks2| jd dkr>td n| jd dkrzddlm} W n tk
rz   td	  Y nX t| jd
krtd n*| j }| j }| j }|||| n | jd dkrJddlm	} t| jdk st| jd
krtd n@| jd }| jd }t| jd
kr6| jd }nd}|||| n| jd dkr$zddl
m} W n  tk
r   td	  Y nX t| jdk st| jd
krtd njd| jd  kr| jd }d}	n| jd }	d}| jd }t| jd
kr| jd }nd}|||||	 n| jd dkrzddlm}
 W n  tk
rd   td	  Y nX t| jdk st| jd
krtd n@| jd }| jd }t| jd
kr| jd }nd}|
||| n| jd dkrzddlm} W n  tk
r   td	  Y nX t| jd
k s6t| jdkr@td nL| jd }| jd }| jd }t| jdkrz| jd
 }nd }||||| nT| jd dkrddlm} t| jdkrtd n| jd }| jd }||| d S )N             )bertgpt
transfo_xlgpt2xlnetxlma\  This command line utility let you convert original (author released) model checkpoint to pytorch.
It should be used as one of: 
>> transformers bert TF_CHECKPOINT TF_CONFIG PYTORCH_DUMP_OUTPUT, 
>> transformers gpt OPENAI_GPT_CHECKPOINT_FOLDER_PATH PYTORCH_DUMP_OUTPUT [OPENAI_GPT_CONFIG], 
>> transformers transfo_xl TF_CHECKPOINT_OR_DATASET PYTORCH_DUMP_OUTPUT [TF_CONFIG] or 
>> transformers gpt2 TF_CHECKPOINT PYTORCH_DUMP_OUTPUT [GPT2_CONFIG] or 
>> transformers xlnet TF_CHECKPOINT TF_CONFIG PYTORCH_DUMP_OUTPUT [FINETUNING_TASK_NAME] or 
>> transformers xlm XLM_CHECKPOINT_PATH PYTORCH_DUMP_OUTPUTr   ) convert_tf_checkpoint_to_pytorchztransformers can only be used from the commandline to convert TensorFlow models in PyTorch, In that case, it requires TensorFlow to be installed. Please see https://www.tensorflow.org/install/ for installation instructions.   zQShould be used as `transformers bert TF_CHECKPOINT TF_CONFIG PYTORCH_DUMP_OUTPUT`r   )$convert_openai_checkpoint_to_pytorchznShould be used as `transformers gpt OPENAI_GPT_CHECKPOINT_FOLDER_PATH PYTORCH_DUMP_OUTPUT [OPENAI_GPT_CONFIG]`       r   )(convert_transfo_xl_checkpoint_to_pytorchziShould be used as `transformers transfo_xl TF_CHECKPOINT/TF_DATASET_FILE PYTORCH_DUMP_OUTPUT [TF_CONFIG]`Zckptr   )"convert_gpt2_checkpoint_to_pytorchzSShould be used as `transformers gpt2 TF_CHECKPOINT PYTORCH_DUMP_OUTPUT [TF_CONFIG]`r	   )#convert_xlnet_checkpoint_to_pytorchziShould be used as `transformers xlnet TF_CHECKPOINT TF_CONFIG PYTORCH_DUMP_OUTPUT [FINETUNING_TASK_NAME]`r
   )!convert_xlm_checkpoint_to_pytorchzLShould be used as `transformers xlm XLM_CHECKPOINT_PATH PYTORCH_DUMP_OUTPUT`)syslenargvprintZ.convert_bert_original_tf_checkpoint_to_pytorchr   ImportErrorpopZ0convert_openai_original_tf_checkpoint_to_pytorchr   Z4convert_transfo_xl_original_tf_checkpoint_to_pytorchr   lowerZ.convert_gpt2_original_tf_checkpoint_to_pytorchr   Z/convert_xlnet_original_tf_checkpoint_to_pytorchr   Z2convert_xlm_original_pytorch_checkpoint_to_pytorchr   )r   r   ZPYTORCH_DUMP_OUTPUTZ	TF_CONFIGZTF_CHECKPOINTr   Z!OPENAI_GPT_CHECKPOINT_FOLDER_PATHZOPENAI_GPT_CONFIGr   ZTF_DATASET_FILEr   r   ZFINETUNING_TASKr   ZXLM_CHECKPOINT_PATH r   9/tmp/pip-unpacked-wheel-ymerj3tt/transformers/__main__.pymain   s    *







 



 


 






r   __main__N)r   __name__r   r   r   r   <module>   s   ~