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 �EgD�@s4dd
lm
Z

ddlmZ
dd�ZGdd�de�ZdS) �)
�ArgumentParser�
)
�BaseTransformersCLICommandc

Cst|j
|j|j�S)
N)�DownloadCommand�model� cache_dir�force)
�args�r �H/usr/local/lib/python3.6/site-packages/transformers/commands/download.py�download_command_factorys
rc@s8eZ
dZeed
�
dd��
Zeeed�dd�Zdd�Z d S) r)
�parserc
CsL|jd
�
}
|
j
dtddd�
|
j
dddd�
|
j
d td d�
|
jtd�

dS) N�downloadz--cache-dirz$Path to location to store the models)�type�default�helpz--force� store_truez;Force the model to be download even if already in cache-dir)�actionrrzName of the model to download)rr)
�func)� add_parser�add_argument�str�set_defaultsr)r Zdownload_parserr r r�register_subcommands



z#DownloadCommand.register_subcommand)r�cachercCs|
|_||_
||_dS)
N)�_model�_cache�_force)�selfrrrr r r�__init__%s


zDownloadCommand.__init__c
Cs@d
dlm
}
m}
|
j|j|j|jd�
|j|j|j|jd�
dS)N�)� AutoModel� AutoTokenizer)rZforce_download)Zmodels.autor!r"Zfrom_pretrainedrrr)rr!r"r r r�run*s

zDownloadCommand.runN) �__name__� __module__�__qualname__�staticmethodrrr�boolrr#r r r rrs

rN)�argparser�rrrr r r r�s