hCLSrSSKrSSKrSSKrSSKrSSKrSSKJr SSKJ r SSK J r J r J r JrJrJrJrJr SSKJr SSKJr SSKrS\R0;a\ "\R0SS 5qOLS \R0;a\ "\R0S S S 5qO$\ "\R4R7S 5S S S 5q"SS5rS\SS4SjrS\4SjrS\S\\\\44Sjr S\S\!4Sjr"S*S\S\!S\4Sjjr#S\S\SS4Sjr$S+S\S\\S\\\S\\4Sjjr%S,S\S\S\\&S\4S jjr'S-S\S\S\\&S!\!S"\!S\\RP4 S#jjr)S-S\S\S\\&S!\!S"\!S\\4 S$jjr*S.S%\S&\S'\S(\S\\&S!\!S"\!S\\RP4S)jjr+g)/zJONNX Model Hub This implements the python client for the ONNX model hub. N)BytesIO)join)IOAnyDictListOptionalSetTuplecast) HTTPError)urlopen ONNX_HOMEhubXDG_CACHE_HOMEonnx~z.cachecP\rSrSrSrS\\\4SS4SjrS\4Sjr S\4Sjr S r g) ModelInfoa* A class to represent a model's property and metadata in the ONNX Hub. It extracts model name, path, sha, tags, etc. from the passed in raw_model_info dict. Attributes: model: The name of the model. model_path: The path to the model, relative to the model zoo (https://github.com/onnx/models/) repo root. metadata: Additional metadata of the model, such as the size of the model, IO ports, etc. model_sha: The SHA256 digest of the model file. tags: A set of tags associated with the model. opset: The opset version of the model. raw_model_inforeturnNc[[US5Ul[[US5Ul[[[[ 4US5UlSUlSUR ;a"[[UR S5Ul[5Ul SUR ;a2[[[[UR S55Ul [[US5Ul Xl g)zH Parameters: raw_model_info: A JSON dict containing the model info. model model_pathmetadataN model_shatags opset_version)r strrrrrrrsetrrintopsetr)selfrs B/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/hub.py__init__ModelInfo.__init__+s #~g67 sN<$@A(,T#s(^^J=W(X (, $-- '!#t}}['ABDN!e T]] "DcDMM&,ABCDI#~o>? .<c pSURSURSURSURS3 $)NzModelInfo(model=z, opset=z, path=z , metadata=))rr#rrr$s r%__str__ModelInfo.__str__?s:!$**Xdjj\HYYdeiererdsstuur(c"UR5$N)r,r+s r%__repr__ModelInfo.__repr__Bs||~r()rrrrr#rr) __name__ __module__ __qualname____firstlineno____doc__rr rr&r,r0__static_attributes__r(r%rrs< =tCH~=$=(vv#r(rnew_dirrcUqg)z[ Sets the current ONNX hub cache location :param new_dir: location of new model hub cache N _ONNX_HUB_DIR)r9s r%set_dirr=Fs Mr(c[$)z] Gets the current ONNX hub cache location :return: The location of the ONNX hub model cache r;r8r(r%get_dirr?Ps r(repocURS5SRS5SnURS5SRS5SnSU;aURS5SnOSnXU4$)zE Gets the repo owner, name and ref from a repo specification string. :r/main)splitr@ repo_owner repo_namerepo_refs r%_parse_repo_inforKYsqC#))#.q1J 3"((-a0I d{::c?1% ( **r(cX[U5upnUS:H=(a US:H=(a US:H$)zs Verifies whether the given model repo can be trusted. A model repo can be trusted if it matches onnx/models:main. rmodelsrErKrGs r%_verify_repo_refrOfs2 '7t&<#J8 & VyH'< V8vCUVr(lfsc\[U5up#nU(a SUSUSUS3$SUSUSUS3$)a  Gets the base github url from a repo specification string :param repo: The location of the model repo in format "user/repo[:branch]". If no branch is found will default to "main" :param lfs: whether the url is for downloading lfs models :return: the base github url for downloading z*https://media.githubusercontent.com/media/rCz"https://raw.githubusercontent.com/rN)r@rPrHrIrJs r% _get_base_urlrRosQ'7t&<#J8 ;J get_model_info..s r()keyz has no version with opset z. Valid opsets: r)rrrbsortedr#)rr@r#rmselected_modelsrl valid_opsetss r%get_model_infor{s"$.O 9%ABB } 6HI 1  '6JoE9I1oJ-<=_GG_L= '4UG;KL>Z  1  K=sBB"B force_reloadsilentc [XU5nURRS5nURbURSUS3US'[ [ [ RR U55nU(d$[ RRU5(d[U5(dMU(dFSUS3n[U[RS9 [S5 [5R5S :wag[ R "[ RR#U5S S 9 [%US 5n [S US U35 ['XR-U5 O[SUSU35 [)US5n U R+5n SSS5 URb\[,R."W 5R15n XR:Xd)[3SUR4SU SURS35e[6R8"[;[<[>[AW 555$!,(df  N=f)aM Downloads a model by name from the onnx model hub :param model: The name of the onnx model in the manifest. This field is case-sensitive :param repo: The location of the model repo in format "user/repo[:branch]". If no branch is found will default to "main" :param opset: The opset of the model to download. The default of `None` automatically chooses the largest opset :param force_reload: Whether to force the model to re-download even if its already found in the cache :param silent: Whether to suppress the warning message if the repo is not trusted. :return: ModelProto or None rCN_The model repo specification _ is not trusted and may contain security vulnerabilities. Only continue if you trust this repo.fileContinue?[y/n]yTexist_ok Downloading  to local path Using cached model from rbThe cached model has SHA256  while checksum should be h. The model in the hub may have been updated. Use force_reload to download the model from the model hub.)!r{rrFrrr<osseppathexistsrOprintsysstderrinputrcmakedirsdirnamerRr^rWrXhashlibsha256 hexdigestrbrrrar rbytesr) rr@r#r|r}selected_modellocal_model_path_arrlocal_model_pathmsglfs_urlr\ model_bytesdownloaded_shas r%raras$$E7N)44::3?+''(*>r*B)C D   M266;;7K+LM277>>*:;;%%f1$8WXC #CJJ ' " #w}}#% BGGOO$45Ed+ UG?3C2DEFw!:!::>@!9!99 #N$8$8#9nEUV,,:,D,D+EF99  99T"U)W[%9: ;; & %s H<< I c [XU5nURSRS5nURSnUb USUS3US'[[[ R RU55nU(d$[ RRU5(d[U5(dMU(dFSUS3n [U [RS 9 [S 5 [5R5S :wag[ R"[ RR!U5S S 9 [#US 5n [SUSU35 [%XRS-U5 O[SUSU35 ['US5n U R)5n SSS5 UbH[*R,"W 5R/5n X:Xd[1SUR2SU SUS35e[4R&"U5nUS[7U5S- nUR9U5 SSS5 WS-[ R:"U5S-nU$!,(df  N=f!,(df  N@=f)a Downloads a model along with test data by name from the onnx model hub and returns the directory to which the files have been extracted. :param model: The name of the onnx model in the manifest. This field is case-sensitive :param repo: The location of the model repo in format "user/repo[:branch]". If no branch is found will default to "main" :param opset: The opset of the model to download. The default of `None` automatically chooses the largest opset :param force_reload: Whether to force the model to re-download even if its already found in the cache :param silent: Whether to suppress the warning message if the repo is not trusted. :return: str or None model_with_data_pathrCmodel_with_data_shaNrrrrrrrTrrrrrrrrrrr)r{rrFrr<rrrrrOrrrrrcrrrRr^rWrXrrrrbrtarfilerd extractalllistdir)rr@r#r|r}rlocal_model_with_data_path_arrrlocal_model_with_data_pathrrr\model_with_data_bytesrmodel_with_data_zippedlocal_model_with_data_dir_pathrs r%download_model_with_test_datarsj$$E7N%3%<%<& eCj#)112GH&##1%CB%G$H I '  "&rvv{{#AB"277>>*DEE%%f1$8WXC #CJJ ' " #w}}#% BGGOO$>?$Od+ UG?3M2NOP --.DE E &  eWL1K0LMN ($ /1 ! 0& (=>HHJ4 #N$8$8#9nEUV,,?+@A99  0 15K)C ./!3* & ))*HI 2 '   **3 4Q 7 8 3 0 / 2 1sI #I I I, network_modelpreprocessing_model network_repopreprocessing_repoc[XXEU5nUc[SU35e[XXEU5nUc[SU35e[5n 0n 0n URH@n U RS:XaSO U Rn U R U 5 U R X'MB URH@n U RS:XaSO U Rn U R U 5 U R X'MB SnSnU HJn U S:Xa XnXnMX;dMX;dM!XX:wdM-[SUSUSU S XS XS 3 5e X:aP[RRX5nURUl [RRU5 OTX:aO[RRX5nURUl [RRU5 [URR URR"5VVs/sHunnUR$UR$4PM nnn[R&R)XxUS 9nU$s snnf) a Builds a composite model including data preprocessing by downloading a network and a preprocessing model and combine it into a single model :param model: The name of the onnx model in the manifest. This field is case-sensitive :param repo: The location of the model repo in format "user/repo[:branch]". If no branch is found will default to "main" :param opset: The opset of the model to download. The default of `None` automatically chooses the largest opset :param force_reload: Whether to force the model to re-download even if its already found in the cache :param silent: Whether to suppress the warning message if the repo is not trusted. :return: ModelProto or None z(Could not load the preprocessing model: z"Could not load the network model: zai.onnxrzCan not merge z and z: because they contain different opset versions for domain z (z) and z\). Only the default domain can be automatically converted to the highest version of the two.)io_map)ra RuntimeErrorr! opset_importdomainaddversion ValueErrorrversion_converterconvert_version ir_versionchecker check_modelzipgraphoutputrnamecompose merge_models)rrrrr#r|r} preprocessingnetwork all_domainsdomains_to_version_network domains_to_version_preprocessingopset_import_entryrpreprocessing_opset_versionnetwork_opset_version out_entryin_entryrmodel_with_preprocessings r%load_composite_modelrcs*fM67J6K L  =VLG? OPPEK1379$%22+22b8I>P>W>W  -?-G-G"* 3,88+22b8I>P>W>W  3E3M3M(0 9#% Y *J*R '$>$F !  6408/78 !4 5U=/J77=hbAaAi@jk19:;MM  #:((88  +55   )  <..>>  $+#5#5     /$'}':':'A'A7==CVCV#W#W Ix '#W  $||88v 9  $#s%I')F)onnx/models:mainNN)rN)rNFF)rrNFF),r6rr`rrrioros.pathrtypingrrrrr r r r urllib.errorr urllib.requestrrenvironr<r expanduserrr r=r?rKboolrOrRr^rrr"r{ ModelProtorarrr8r(r%rs BBB"" "**K0%8M#$45vuEM++C0(FEJM&&RST +3 +5c3#7 +W3W4W U U$ U3 U $# $) ) C=) 49 ) )_ )ZHL 7?}<# 7< 7< 7< C=7< 7<  7< doo 7