L i[ 8ddlmZddlmZdZGddeZy))ArgumentParser)BaseTransformersCLICommandcnt|j|j|j|jSN)DownloadCommandmodel cache_dirforcetrust_remote_code)argss d/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/transformers/commands/download.pydownload_command_factoryrs% 4::t~~tzz4CYCY ZZc@eZdZedefdZdedededefdZdZ y ) rparserc|jd}|jdtdd|jddd|jd dd |jd td |jty)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 machiner zName of the model to download)rr)func) add_parser add_argumentstr set_defaultsr)rdownload_parsers rregister_subcommandz#DownloadCommand.register_subcommands ++J7$$ T8^ %  $$ l1n %  $$ !{ % $$W3=\$]$$*B$Crr cacher r c<||_||_||_||_yr)_model_cache_force_trust_remote_code)selfr r!r r s r__init__zDownloadCommand.__init__*s   "3rcddlm}m}|j|j|j |j |j|j|j|j |j |jy)N) AutoModel AutoTokenizer)r force_downloadr ) models.autor+r,from_pretrainedr#r$r%r&)r'r+r,s rrunzDownloadCommand.run0se:!! KK4;;t{{^b^u^u "  %% KK4;;t{{^b^u^u & rN) __name__ __module__ __qualname__ staticmethodrr rboolr(r0rrrrsDDNDD 4c4#4d4t4  rrN)argparserrrrr6rrr9s $([  0  r