o jC@sdZddlZddlmZmZddlZddlmZmZm Z ddl m Z m Z ddl mZddlmZmZmZmZmZmZmZmZmZmZmZdd lmZdd lmZeee Z ee Z!ej"d d d e!De#dZ$ee#dBej%dd&e dee dfZ'eddZ(e(j)dgddddddddeddddddf dee#dBej*ddfdedededee#dBej%ddfd ee$dBej%d!dfd"ed#e'd$ee+ej%d%d&d'dfd(ee+ej%d)d*dfd+ee+ej%d,d-d.dfd/ed0ed1dfd2d3Z,e(j)d4d5d6gd   dEd7ee#ej*d8dfd/ed#e'd0ed1df d9d:Z-e(j)d;gd<d   dFd7ee#ej*d8dfd=ee+ej%d>d?dfd@ee+ej%dAdBdfd0ed1df dCdDZ.dS)GaContains 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) Annotatedget_args)CLIErrorRepositoryNotFoundErrorRevisionNotFoundError)ExpandModelProperty_T ModelSort_T) ModelCard) REPO_LIST_DEFAULT_LIMIT AuthorOpt FilterOptLimitOpt RevisionOpt SearchOptTokenOptapi_object_to_dict get_hf_apimake_expand_properties_parser typer_factory)list_repo_files_cmd)out ModelSortEnumcCsi|]}||qSr).0srrM/home/kuhnn/.local/lib/python3.10/site-packages/huggingface_hub/cli/models.py 7sr)typezComma-separated properties to return. When used, only the listed properties (and id) are returned. Example: '--expand=downloads,likes,tags'. Valid: z, .)helpcallbackz Interact with models on the Hub.)r z list | ls)z(hf models ls --sort downloads --limit 10z1hf models ls --search "llama" --author meta-llamaz:hf models ls --num-parameters min:6B,max:128B --sort likesz-hf models ls meta-llama/Llama-3.2-1B-Instructz0hf models ls meta-llama/Llama-3.2-1B-Instruct -Rz7hf models ls meta-llama/Llama-3.2-1B-Instruct --tree -h)examplesFrepo_idzRModel ID (e.g. `username/repo-name`) to list files from. If omitted, lists models.searchauthorfilternum_parametersz2Filter by parameter count, e.g. 'min:6B,max:128B'.sortz Sort results.limitexpandhuman_readablez--human-readablez-hz=Show sizes in human readable format (only for listing files).as_treez--treez3List files in tree format (only for listing files). recursivez --recursivez-Rz0List files recursively (only for listing files).revisiontokenreturnc  Cs |durN|dur td|durtd|durtd|dur(td|dur1td|tkr:td|durCtdt|d || | | | d S| rUtd | r\td |rctd | durltdt| d} |rv|jnd}dd| j|||||||dD}t|dS)zList models on the Hub, or files in a model repo. When called with no argument, lists models on the Hub. When called with a model ID, lists files in that model repo. Nz'Cannot use --search when listing files.z'Cannot use --author when listing files.z'Cannot use --filter when listing files.z/Cannot use --num-parameters when listing files.z%Cannot use --sort when listing files.z&Cannot use --limit when listing files.z'Cannot use --expand when listing files.model)r# repo_typer+r,r-r.r/z&Cannot use --tree when listing models.z+Cannot use --recursive when listing models.z0Cannot use --human-readable when listing models.z*Cannot use --revision when listing models.r/cSsg|]}t|qSr)r)r model_inforrr szmodels_ls..)r&r%r$r'r(r)r*) typer BadParameterr rrvalue list_modelsrtable)r#r$r%r&r'r(r)r*r+r,r-r.r/apisort_keyresultsrrr models_lsFs\1             r>infoz/hf models info meta-llama/Llama-3.2-1B-InstructzrCrLrrrrs   4         X