o  h[ @s4ddlmZddlmZddZGdddeZdS) )ArgumentParser)BaseTransformersCLICommandcCst|j|j|j|jSN)DownloadCommandmodel cache_dirforcetrust_remote_code)argsr r/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/transformers/commands/download.pydownload_command_factorysrc@s@eZdZedefddZdedededefdd Zd d Z d S) rparsercCs\|d}|jdtddd|jdddd|jd dd d|jd td d |jtddS)Ndownloadz --cache-dirz$Path to location to store the models)typedefaulthelpz--force store_truez;Force the model to be download even if already in cache-dir)actionrz--trust-remote-codezWhether or not to allow for custom models defined on the Hub in their own modeling files. Use only if you've reviewed the code as it will execute on your local machinerzName of the model to download)rr)func) add_parser add_argumentstr set_defaultsr)rdownload_parserr r r register_subcommands z#DownloadCommand.register_subcommandrcacher r cCs||_||_||_||_dSr)_model_cache_force_trust_remote_code)selfrrr r r r r __init__*s zDownloadCommand.__init__cCsHddlm}m}|j|j|j|j|jd|j|j|j|j|jddS)N) AutoModel AutoTokenizer)rforce_downloadr ) models.autor%r&from_pretrainedrrr r!)r"r%r&r r r run0s zDownloadCommand.runN) __name__ __module__ __qualname__ staticmethodrrrboolr#r*r r r r rs  rN)argparserrrrr r r r s