a
    dD                     @   s4   d dl mZ ddlmZ dd ZG dd deZdS )	    )ArgumentParser   )BaseTransformersCLICommandc                 C   s   t | j| j| jS N)DownloadCommandmodel	cache_dirforce)args r   g/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/commands/download.pydownload_command_factory   s    r   c                   @   s8   e Zd ZeedddZeeedddZdd Z	d	S )
r   )parserc                 C   sL   |  d}|jdtd dd |jdddd |jd	td
d |jtd d S )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)actionr   r   zName of the model to download)r   r   )func)
add_parseradd_argumentstrset_defaultsr   )r   Zdownload_parserr   r   r   register_subcommand   s    
z#DownloadCommand.register_subcommand)r   cacher	   c                 C   s   || _ || _|| _d S r   )_model_cache_force)selfr   r   r	   r   r   r   __init__%   s    zDownloadCommand.__init__c                 C   s@   ddl m}m} |j| j| j| jd |j| j| j| jd d S )N   )	AutoModelAutoTokenizer)r   Zforce_download)Zmodels.autor"   r#   Zfrom_pretrainedr   r   r   )r   r"   r#   r   r   r   run*   s    zDownloadCommand.runN)
__name__
__module____qualname__staticmethodr   r   r   boolr    r$   r   r   r   r   r      s   r   N)argparser    r   r   r   r   r   r   r   <module>   s   