fThn SSKrSSKJrJr SSKJrJrJrJrJ r J r J r SSK r SSK JrJrJr SSKJrJrJr SSKJr \(aSS K JrJr \R2"\5r\"5(a"SS KJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r*J+r+ \"5(aSS KJ,r,J-r-J.r.J/r/J0r0J1r1J2r2J3r3J4r4 \"5(d\"5(d\RkS 5 SS .S\6S\\6S\\6\\/\444Sjjr7"SS5r8g)N)partialreduce) TYPE_CHECKINGCallableDictOptionalTupleTypeUnion)PretrainedConfigis_tf_availableis_torch_available)TF2_WEIGHTS_NAME WEIGHTS_NAMElogging) OnnxConfigPreTrainedModelTFPreTrainedModel) AutoModelAutoModelForCausalLMAutoModelForImageClassificationAutoModelForImageSegmentationAutoModelForMaskedImageModelingAutoModelForMaskedLMAutoModelForMultipleChoiceAutoModelForObjectDetectionAutoModelForQuestionAnswering AutoModelForSemanticSegmentationAutoModelForSeq2SeqLM"AutoModelForSequenceClassificationAutoModelForSpeechSeq2SeqAutoModelForTokenClassificationAutoModelForVision2Seq) TFAutoModelTFAutoModelForCausalLMTFAutoModelForMaskedLMTFAutoModelForMultipleChoiceTFAutoModelForQuestionAnswering"TFAutoModelForSemanticSegmentationTFAutoModelForSeq2SeqLM$TFAutoModelForSequenceClassification!TFAutoModelForTokenClassificationzThe ONNX export features are only supported for PyTorch or TensorFlow. You will not be able to export models without one of these libraries installed.onnx_config_clssupported_featuresr1returncUc [S5e[nURS5Hn[X#5nM 0nUHInSU;a*UR SS5n[ UR US9XE'M3[ URUS9XE'MK U$)aG Generate the mapping between supported the features and their corresponding OnnxConfig for a given model. Args: *supported_features: The names of the supported features. onnx_config_cls: The OnnxConfig full name corresponding to the model. Returns: The dictionary mapping a feature to an OnnxConfig constructor. z#A OnnxConfig class must be provided. -with-past)task) ValueError transformerssplitgetattrreplacer with_pastfrom_model_config)r1r2 config_cls attr_namemappingfeaturer8s R/var/www/auris/envauris/lib/python3.13/site-packages/transformers/onnx/features.pysupported_features_mappingrE7s>??J$**3/ Z3 0G% 7 "??<4D&z';';$GG &z'C'C'RG  & Ncb\rSrSr0r0r\"5(a\\\ \ \ \ \ \\\\\\\\S.r\"5(a \\\\\\\\\S. r0S\ "SSSSS S S S 9_S \ "SSSSSSSS SS 9 _S\ "SSSS 9_S\ "SSSSSS S SS 9_S\ "SSSSSS S SS 9_S\ "SSSSSSSS SS 9 _S\ "SSSSSSSS 9_S\ "SSSSSSS S 9_S!\ "SSSSSS S"S 9_S#\ "SSSSSS S S$S 9_S%\ "SS&S 9_S'\ "SSS(S 9_S)\ "SSSSS S S*S 9_S+\ "SSS,S 9_S-\ "SSSSS S S.S 9_S/\ "SSS0S 9_S1\ "SSSS S S2S 9_0S3\ "SSSSS S S4S 9_S5\ "SSS6S 9_S7\ "SS8S9S:S 9_S;\ "SSSSS S SS 9_S?\ "SSSSSS S S@S 9_SA\ "SSSSSS SBS 9_SC\ "SSSSS SSDS 9_SE\ "SSSSSSFS 9_SG\ "SSHS 9_SI\ "SSSSS S SJS 9_SK\ "SSSLS 9_SM\ "SSSS SNS 9_SO\ "SS SS SPS 9_SQ\ "SSSRS 9_SS\ "SSSSSTS 9_SU\ "SSSS SS SVS 9_E0SW\ "SSSSSSSXS 9_SY\ "SSSSSSSS SZS 9 _S[\ "SSSSS S S\S 9_S]\ "SSS^S 9_S_\ "SSS`S 9_Sa\ "SSSbS 9_Sc\ "SSSSSdS 9_Se\ "SSSSSfS 9_Sg\ "SShS 9_Si\ "SSSSjS 9_Sk\ "SSSlS 9_Sm\ "SSSSSS S SnS 9_So\ "SSSpS 9_Sq\ "SSSSSS S SrS 9_Ss\ "SSSSS SS S StS 9 _Su\ "SSSvSwS 9_Sx\ "SSSSS S SyS 9_E\ "SSSzS 9\ "SSSSS{S 9\ "S|S}S 9\ "SSS~S 9\ "SSSSSS 9\ "SSSSSS S SS 9\ "SSSSSS S SS 9\ "SS8SS 9S.Er!\""\#"SS\!RI5555r%\&SS\'S\(\'S\)\'\*\+/\,444Sjj5r-\&S\'S\'4Sj5r.\&S\'4Sj5r/\&SS\'S\'S\04Sjj5r1\&SS\'S\(\'S\'4Sjj5r2\&SS\'S\'S\(\'S\(\'S\3S4 Sjj5r4\&SS\3SS\'S\5\'\*44Sjj5r6S\'S\'S\,4Sjr7Sr8g)FeaturesManagerU)default masked-lm causal-lm seq2seq-lmsequence-classificationtoken-classificationmultiple-choiceobject-detectionquestion-answeringimage-classificationimage-segmentationz masked-imsemantic-segmentation vision2seq-lm speech2seq-lm) rJrKrLrMrNrOrPrRrUalbertrJrKrNrPrOrRzmodels.albert.AlbertOnnxConfigr0bartzdefault-with-pastrLzcausal-lm-with-pastrMzseq2seq-lm-with-pastzmodels.bart.BartOnnxConfigbeitrSzmodels.beit.BeitOnnxConfigbertzmodels.bert.BertOnnxConfigzbig-birdz!models.big_bird.BigBirdOnnxConfigzbigbird-pegasusz/models.bigbird_pegasus.BigBirdPegasusOnnxConfig blenderbotz&models.blenderbot.BlenderbotOnnxConfigzblenderbot-smallz1models.blenderbot_small.BlenderbotSmallOnnxConfigbloomzmodels.bloom.BloomOnnxConfig camembertz$models.camembert.CamembertOnnxConfigclipzmodels.clip.CLIPOnnxConfigcodegenz models.codegen.CodeGenOnnxConfigconvbertz"models.convbert.ConvBertOnnxConfigconvnextz"models.convnext.ConvNextOnnxConfigz data2vec-textz&models.data2vec.Data2VecTextOnnxConfigzdata2vec-visionz(models.data2vec.Data2VecVisionOnnxConfigdebertaz models.deberta.DebertaOnnxConfigz deberta-v2z%models.deberta_v2.DebertaV2OnnxConfigdeitzmodels.deit.DeiTOnnxConfigdetrrQrTzmodels.detr.DetrOnnxConfig distilbertz&models.distilbert.DistilBertOnnxConfigelectraz models.electra.ElectraOnnxConfigflaubertz"models.flaubert.FlaubertOnnxConfiggpt2zmodels.gpt2.GPT2OnnxConfiggptjzmodels.gptj.GPTJOnnxConfigzgpt-neozmodels.gpt_neo.GPTNeoOnnxConfiggroupvitz"models.groupvit.GroupViTOnnxConfigibertzmodels.ibert.IBertOnnxConfigimagegptz"models.imagegpt.ImageGPTOnnxConfiglayoutlmz"models.layoutlm.LayoutLMOnnxConfig layoutlmv3z&models.layoutlmv3.LayoutLMv3OnnxConfiglevitzmodels.levit.LevitOnnxConfiglongt5zmodels.longt5.LongT5OnnxConfig longformerz&models.longformer.LongformerOnnxConfigmarianzmodels.marian.MarianOnnxConfigmbartzmodels.mbart.MBartOnnxConfig mobilebertz&models.mobilebert.MobileBertOnnxConfigz mobilenet-v1z)models.mobilenet_v1.MobileNetV1OnnxConfigz mobilenet-v2z)models.mobilenet_v2.MobileNetV2OnnxConfig mobilevitz$models.mobilevit.MobileViTOnnxConfigmt5zmodels.mt5.MT5OnnxConfigzm2m-100zmodels.m2m_100.M2M100OnnxConfigowlvitzmodels.owlvit.OwlViTOnnxConfig perceiverz$models.perceiver.PerceiverOnnxConfig poolformerz&models.poolformer.PoolFormerOnnxConfigrembertz models.rembert.RemBertOnnxConfigresnetzmodels.resnet.ResNetOnnxConfigrobertaz models.roberta.RobertaOnnxConfigroformerz"models.roformer.RoFormerOnnxConfig segformerrUz$models.segformer.SegformerOnnxConfig squeezebertz(models.squeezebert.SqueezeBertOnnxConfigzmodels.swin.SwinOnnxConfigzmodels.t5.T5OnnxConfigrVzFeaturesManager.*sbgrFc#@# UHoR5v M g7fr)keys).0vs rD FeaturesManager.*s?qRpQRpsN model_type model_namer3cUR5nU[R;aJU(aUSUS3OUn[US[ [RR 55SUS35e[RU$)a Tries to retrieve the feature -> OnnxConfig constructor map from the model type. Args: model_type (`str`): The model type to retrieve the supported features for. model_name (`str`, *optional*): The name attribute of the model object, only used for the exception message. Returns: The dictionary mapping each feature to a corresponding OnnxConfig constructor. z ()z is not supported yet. Only z' are supported. If you want to support z) please propose a PR or open up an issue.)lowerrH_SUPPORTED_MODEL_TYPEKeyErrorlistr)rrmodel_type_and_model_names rD%get_supported_features_for_model_type5FeaturesManager.get_supported_features_for_model_type,s  %%' _BB BJT:,b A(FZd %,-._BBGGIJKL**45^`  44Z@@rFrCc&URSS5$)Nr6r7)r=)rCs rDfeature_to_taskFeaturesManager.feature_to_taskFs|R00rF frameworkcUS;a[SUS35eUS:Xa[5(d [S5eUS:Xa[5(d [S5egg) zs Validates if the framework requested for the export is both correct and available, otherwise throws an exception. pttfzAOnly two frameworks are supported for ONNX export: pt or tf, but z was provided.rzOCannot export model to ONNX using PyTorch because no PyTorch package was found.rzUCannot export model to ONNX using TensorFlow because no TensorFlow package was found.N)r9r RuntimeErrorr)rs rD_validate_framework_choice*FeaturesManager._validate_framework_choiceJsk L (ST]S^^lm $ '9';';pq q $ '8'8vw w(9 rFc "[RU5n[RU5 US:Xa[RnO[RnX#;a6[ SUS[ [RR5535eX2$)a Attempts to retrieve an AutoModel class from a feature name. Args: feature (`str`): The feature required. framework (`str`, *optional*, defaults to `"pt"`): The framework to use for the export. Returns: The AutoModel class corresponding to the feature. rzUnknown task: z. Possible values are )rHrr_TASKS_TO_AUTOMODELS_TASKS_TO_TF_AUTOMODELSrrvalues)rCrr8task_to_automodels rDget_model_class_for_feature+FeaturesManager.get_model_class_for_featureYs..w7229=   / D D  / G G   (  )?_EiEiEpEpEr@s?tu !&&rFmodelcUbU$SSS.nSSS.n[RRU5(a[RR[RR U[ 55(aSnOc[RR[RR U[ 55(aSnO[S[ S [ S 35e[RS X!S 35 O/[5(aSnO[5(aSnO [S 5e[RSX1S35 U$)a Determines the framework to use for the export. The priority is in the following order: 1. User input via `framework`. 2. If local checkpoint is provided, use the same framework as the checkpoint. 3. Available framework in environment, with priority given to PyTorch Args: model (`str`): The name of the model to export. framework (`str`, *optional*, defaults to `None`): The framework to use for the export. See above for priority if none provided. Returns: The framework to use for the export. PyTorch TensorFlowrtorchtf2onnxrrzMCannot determine framework from given checkpoint location. There should be a z for PyTorch or z for TensorFlow.zLocal z model found.zKNeither PyTorch nor TensorFlow found in environment. Cannot export to ONNX.zFramework not requested. Using z to export to ONNX.) ospathisdirisfilejoinrrFileNotFoundErrorloggerinforrEnvironmentError)rr framework_map exporter_maps rDdetermine_framework#FeaturesManager.determine_frameworkts(   ( = %Y7 77==  ww~~bggll5,?@@  U4D EFF '**68+,,<> KK&!9 :-H I!##  "" &'tuu 5l6M5NNabcrF cache_dirrcF[RX5n[RX5nURXS9nU$![aY US:Xa)[ R S5 URUSUS9nU$[ R S5 URUSUS9nU$f=f)a Attempts to retrieve a model from a model's name and the feature to be enabled. Args: feature (`str`): The feature required. model (`str`): The name of the model to export. framework (`str`, *optional*, defaults to `None`): The framework to use for the export. See `FeaturesManager.determine_framework` for the priority should none be provided. Returns: The instance of the model. )rrz=Loading TensorFlow model in PyTorch before exporting to ONNX.T)from_tfrz=Loading PyTorch model in TensorFlow before exporting to ONNX.)from_ptr)rHrrfrom_pretrainedOSErrorrr)rCrrr model_classs rDget_model_from_feature&FeaturesManager.get_model_from_features($77I %AA'U  ^///KE  ^D  [\#33E4S\3]  [\#33E4S\3]  ^s=6B 6&B B c>URRRSS5n[USS5n[R X#S9nX;a'[ URRSUSU35eURR[RUU4$)a Check whether or not the model has the requested features. Args: model: The model to export. feature: The name of the feature to check if it is available. Returns: (str) The type of the model (OnnxConfig) The OnnxConfig instance holding the model export properties. _-namer7)rz doesn't support feature z. Supported values are: )configrr=r<rHrr9r)rrCrrmodel_featuress rDcheck_supported_model_or_raise.FeaturesManager.check_supported_model_or_raises\\,,44S#> UFB/ (NNzNq  (<<**++DWIMefteuv ||&&(M(Mj(YZa(bbbrFc.[RUU$)a  Gets the OnnxConfig for a model_type and feature combination. Args: model_type (`str`): The model type to retrieve the config for. feature (`str`): The feature to retrieve the config for. Returns: `OnnxConfig`: config for the combination )rHr)rrCs rD get_configFeaturesManager.get_configs44Z@IIrFrr)r)NN)rJ)9__name__ __module__ __qualname____firstlineno__rrrrrrr"r#r%rrr rrrr!r&r$rr'r)r(r-r.r/r*r+r,rErsortedrrAVAILABLE_FEATURES staticmethodstrrrrr rrrrr rrr rr rr__static_attributes__rrFrDrHrHUs   --/'I$C9 ;"?$C"?8%E36 ""//1'K$E;"A%G # p,   %  " < p *    !  " % 8  p, * -?[ -p2 *    %  " 8  3pF .    %  " ?  GpZ 5    !  " % M  [pp 0    !  "D qpB 6    !  "O CpT +    ! % ": Upf /    %  " B  gpz * 8 {pB -  > CpL .   %  " @ Mp^ .  "@ _ph 3   %  " D ipz 5  "G  {pH -   % " >  IpX 0   %  " C Ypj * -?[ kpp *   8  qp| 0   %  " D }pN -    %  " >  Opb .    %  " @  cpv *    ! % "8 wpH *    ! %8 IpZ -    ! %=  [pj . @ kpr +   %  " : spD . -?c EpJ .   % "@  KpX 0  % "D  Ypf + -?] gpl ,    "<  mpz 0    % "D {pL ,    "  !< Mp^ +    !  " % :  _pt 0   %  " D upF 2  "G G pP 2  "G Q pZ /  "B [ pd )    "6  e pr -    "=  s p@ , < A pH / "  %B  I pT 0 -?g U pZ -    %  " >  [ pn ,  "< o px -    %  " >  y pL .    % "  "@  M pb /  " #B  c pn 1   %  " F o p@ + -?[ )    "4  #= -k# * -?Y .    %>  *    %  " 6  2    %  " E  ,  : W pd  '=?qRgRnRnRp?q rs59AA%-c]A c8-. :;; <AA21111 xc x x'S'S'D''4/3/8C=/C//b^b -5c]NVWZm 5 6@MVc;<cGJc sH} cc0 Js JS JZ JrFrH)9r functoolsrrtypingrrrrr r r r:r7r rrutilsrrrrrrr get_loggerrrtransformers.models.autorrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/warningrrErHrrFrDrs %NNNDD;;?   H %"   O$5$5 NN 5@D/7} #x)*J67 78