Name	Size	Mode	Actions
add_new_model.cpython-36.pyc	6531	0644	editdlrm
add_new_model_like.cpython-36.pyc	42688	0644	editdlrm
convert.cpython-36.pyc	4943	0644	editdlrm
download.cpython-36.pyc	1660	0644	editdlrm
env.cpython-36.pyc	2605	0644	editdlrm
lfs.cpython-36.pyc	7244	0644	editdlrm
run.cpython-36.pyc	3372	0644	editdlrm
serving.cpython-36.pyc	6748	0644	editdlrm
train.cpython-36.pyc	4589	0644	editdlrm
transformers_cli.cpython-36.pyc	1261	0644	editdlrm
user.cpython-36.pyc	8453	0644	editdlrm
__init__.cpython-36.pyc	794	0644	editdlrm

3 �Eg��@sNdd
lm
Z
mZ
ddlmZ
ddlmZ
ed�
dd�Zd ZGd d�de�Z dS) �)�ArgumentParser� Namespace�)
�logging�
)
�BaseTransformersCLICommand)
�argsc

Cst|j
|j|j|j|j�S)
zx Factory function used to convert a model TF 1.0 checkpoint in a PyTorch checkpoint. Returns: ServeCommand )�ConvertCommand� model_type� tf_checkpoint�pytorch_dump_output�config�finetuning_task_name)
r�r�G/usr/local/lib/python3.6/site-packages/transformers/commands/convert.py�convert_command_factorys
rz� transformers 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. c@s<eZ
dZeed
�
dd��
Zeeeeed�dd�Zdd�Zd S) r )
�parserc
Csx|jd
dd�}
|
j
dtddd�
|
j
dtdd d�
|
j
d tddd�
|
j
dtd dd�
|
j
dtddd�
|
jtd�

dS)z� Register this command to argparse so it's available for the transformer-cli Args: parser: Root parser to register command-specific arguments �convertzcCLI tool to run convert model from original author checkpoints to Transformers PyTorch checkpoints.)
�helpz--model_typeTz Model's type.)�type�requiredrz--tf_checkpointz%TensorFlow checkpoint path or folder.z--pytorch_dump_outputz'Path to the PyTorch saved model output.z--config�z"Configuration file path or folder.)r�defaultrz--finetuning_task_nameNzEOptional fine-tuning task name if the TF model was a finetuned model.)
�func)� add_parser�add_argument�str�set_defaultsr)rZtrain_parserrrr�register_subcommand's









z"ConvertCommand.register_subcommand)r rrr rcGs@tj
d
�
|_|jjd|
���

|
|_||_||_||_||_dS)Nztransformers-cli/convertingzLoading model ) r� get_logger�_logger�info�_model_type�_tf_checkpoint�_pytorch_dump_output�_config�_finetuning_task_name)�selfr rrr rrrrr�__init__Ds 




zConvertCommand.__init__c
 Cs|jd
krNyddl
m}

Wntk r6


tt�
�
YnX|
|j|j|j�
�n�|jdkr�yddlm}

Wntk r�


tt�
�
YnX|
|j|j|j�
�nn|jdkr�yddl m}

Wntk r�


tt�
�
YnX|
|j|j|j�
�n |jdk�
rmodels.albert.convert_albert_original_tf_checkpoint_to_pytorchr)�ImportError�IMPORT_ERROR_MESSAGEr#r%r$Z:models.bert.convert_bert_original_tf_checkpoint_to_pytorchZ>models.funnel.convert_funnel_original_tf_checkpoint_to_pytorchZ6models.t5.convert_t5_original_tf_checkpoint_to_pytorchZ>models.openai.convert_openai_original_tf_checkpoint_to_pytorchr*ZFmodels.transfo_xl.convert_transfo_xl_original_tf_checkpoint_to_pytorchr+�lowerZ:models.gpt2.convert_gpt2_original_tf_checkpoint_to_pytorchr,Zs