aiM dZddlZddlZddlmZmZmZddlZddlm Z m Z ddl m Z m Z ddlmZddlmZmZmZmZmZmZmZmZmZmZeee Zee Zej<d eDcic]}||c}e Z eeeejBd d jEed eefZ#edZ$e$jKd d!dedededeee ejBdfdede#deddfdZ&e$jKd d"deeejNdfdede#deddf d Z(ycc}w)#aContains commands to interact with models on the Hugging Face Hub. Usage: # list models on the Hub hf models ls # list models with a search query hf models ls --search "llama" # get info about a model hf models info Lightricks/LTX-2 N) AnnotatedOptionalget_args)RepositoryNotFoundErrorRevisionNotFoundError)ExpandModelProperty_T ModelSort_T)ANSI) AuthorOpt FilterOptLimitOpt RevisionOpt SearchOptTokenOpt get_hf_apimake_expand_properties_parserrepo_info_to_dict typer_factory ModelSortEnum)typezWComma-separated properties to expand. Example: '--expand=downloads,likes,tags'. Valid: z, .)helpcallbackz Interact with models on the Hub.)rlssearchauthorfiltersortz Sort results.limitexpandtokenreturnct|}|r |jnd}|j||||||D cgc] } t| } } t t j | dycc} w)zList models on the Hub.r"N)rrrrr r!indent)rvalue list_modelsrprintjsondumps) rrrrr r!r"apisort_key model_inforesultss d/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/huggingface_hub/cli/models.py models_lsr3Dsu 5 !C!tzztH//&he\b*   *%G  $**WQ '( sA+infomodel_idz)The model ID (e.g. `username/repo-name`).revisionc t|} |j|||}ttjt|d y #t$r9tdt j |dt jdt$rXtdt j t|dt j |d t jdwxYw) z"Get info about a model on the Hub.r%)repo_idr6r!zModel z not found.r )codez Revision z not found on rr&r'N) rr0rr+r boldtyperExitrstrr,r-r)r5r6r!r"r.r4s r2 models_infor>]s 5 !C!~~h&~Q $**&t,Q 78 #! tyy*+;78jja  ! $))CM23>$))HBUAVVWXYjja  !s A B!C-)NNNN NN)NNN))__doc__enumr,typingrrrr;huggingface_hub.errorsrrhuggingface_hub.hf_apirr huggingface_hub.utilsr _cli_utilsr r rrrrrrrrsorted_EXPAND_PROPERTIES _SORT_OPTIONSEnumr=rOptionjoin ExpandOpt models_clicommandr3Argumentr>)ss0r2rRs  00 QE&   H%:;<%  /-+HQAqD+HsS   SMELLfgkgpgprDhEgFFGH./AB   B C  D ) ) ) )  /* ,  ) ) ) ) ))0 F! 9^U^^1\]]^99 9  9  99S,Is. E