o h@sddlZddlmZddlmZmZddlmZmZm Z ddl m Z ddl m Z Gdd d e ZGd d d Zd eeeeefd eedefddZGdddZGdddeZGdddeZGdddeZGdddeZdS)N)ArgumentParser)ListUnion)HfFolder create_repowhoami) HTTPError)BaseTransformersCLICommandc@seZdZedefddZdS) UserCommandsparsercCs|jddd}|jddd|jddd}|jd dd|jd d d}|jd dd|jd dd}|jdd}|jddd}|jdtdd|jdtdd|jddddd|jddddS)Nloginz6Log in using the same credentials as on huggingface.co)helpcSt|SN) LoginCommandargsrP/var/www/html/ai/venv/lib/python3.10/site-packages/transformers/commands/user.pyz2UserCommands.register_subcommand..)funcrz;Find out which huggingface.co account you are logged in as.cSrr) WhoamiCommandrrrrrrlogoutzLog outcSrr) LogoutCommandrrrrr!rrepoz_Deprecated: use `huggingface-cli` instead. Commands to interact with your huggingface.co repos.zPDeprecated: use `huggingface-cli` instead. huggingface.co repos related commandscreatezNDeprecated: use `huggingface-cli` instead. Create a new repo on huggingface.conamezYName for your model's repo. Will be namespaced under your username to build the model id.)typerz--organizationz!Optional: organization namespace.z-yz--yes store_truez"Optional: answer Yes to the prompt)actionrcSrr)RepoCreateCommandrrrrr5r) add_parser set_defaultsadd_subparsers add_argumentstr)r login_parser whoami_parser logout_parser repo_parserrepo_subparsersrepo_create_parserrrrregister_subcommands0z UserCommands.register_subcommandN)__name__ __module__ __qualname__ staticmethodrr.rrrrr sr c@sDeZdZdZdZdZdZdZeddZ edd Z ed d Z d S) ANSIz; Helper for en.wikipedia.org/wiki/ANSI_escape_code zzzzcC|j||jSr)_bold_resetclssrrrboldBz ANSI.boldcCs|j|j||jSr)r5_redr6r7rrrredFszANSI.redcCr4r)_grayr6r7rrrgrayJr;z ANSI.grayN) r/r0r1__doc__r5r<r>r6 classmethodr:r=r?rrrrr38s  r3rowsheadersreturncCsddtg||RD}dt|j|}g}||j|||jdd|D|D] }||j|q0d|S)z Inspired by: - stackoverflow.com/a/8356620/593036 - stackoverflow.com/questions/9535954/printing-lists-as-tabular-data cSsg|] }tdd|DqS)css|] }tt|VqdSr)lenr').0xrrr Vsz&tabulate...)max)rFcolrrr Vsztabulate..z{{:{}}} cSsg|]}d|qS)-r)rFwrrrrKZs )ziprEformatappendjoin)rBrC col_widths row_formatlinesrowrrrtabulateOs rWc@eZdZddZdS)BaseUserCommandcCs ||_dSrr)selfrrrr__init__as zBaseUserCommand.__init__N)r/r0r1r[rrrrrY` rYc@rX)rcCttddS)NzERROR! `huggingface-cli login` uses an outdated login mechanism that is not compatible with the Hugging Face Hub backend anymore. Please use `huggingface-cli login instead.printr3r=rZrrrrunf zLoginCommand.runNr/r0r1rarrrrrer\rc@rX)rc Csttdt}|durtdtzt|\}}t||r1ttdd|WdSWdSt yW}zt|tt|j j tdWYd}~dSd}~ww)NzxWARNING! `transformers-cli whoami` is deprecated and will be removed in v5. Please use `huggingface-cli whoami` instead. Not logged inzorgs: ,r ) r_r3r=r get_tokenexitrr:rRrresponsetext)rZtokenuserorgserrrraqs* zWhoamiCommand.runNrcrrrrrpr\rc@rX)rcCr])NzERROR! `transformers-cli logout` uses an outdated logout mechanism that is not compatible with the Hugging Face Hub backend anymore. Please use `huggingface-cli logout instead.r^r`rrrrarbzLogoutCommand.runNrcrrrrrr\rc@rX)r"c Csttdt}|durtdtdztddgd}tt | Wn t y9tdYnwztddgd}tt | Wnt y_ttd Ynwtd t |\}}|j jdurt|j jn|}|d |j j}td t||j jstd }|d ks|dks|dkstdtz t||j j|j jd}Wn#ty} zt| tt| jjtdWYd} ~ nd} ~ wwtdtdt|tdtd|td dS)NzlWARNING! Managing repositories through transformers-cli is deprecated. Please use `huggingface-cli` instead.rdr gitz --versionzutf-8z9Looks like you do not have git installed, please install.zgit-lfszLooks like you do not have git-lfs installed, please install. You can install from https://git-lfs.github.com/. Then run `git lfs install` (you only have to do this once)./zYou are about to create zProceed? [Y/n] yyesAbort)r organizationz Your repo now lives at:z zK You can clone it locally with the command below, and commit/push as usual.z git clone )r_r3r=rrfrg subprocess check_outputdecoder?stripFileNotFoundErrorrrrtrr:rrinputlowerrrrhri) rZrjstdoutrk_ namespace full_namechoiceurlrmrrrras`      zRepoCreateCommand.runNrcrrrrr"r\r")ruargparsertypingrrhuggingface_hub.hf_apirrrrequests.exceptionsrror r r3r'intrWrYrrrr"rrrrs   *