mjFUdZddlmZddlZddlmZmZddlmZddl m Z m Z m Z m Z e rddlZeZded<dezezeeeeefzZded <e Zded <Gd d e ZGd de ZGdde ZGdde ZGdde ZdS)z6Typing helpers shared across the Transformers library.) annotationsN)MappingMutableMapping)PathLike) TYPE_CHECKINGAnyProtocol TypeAliasr LevelExcInfoDeviceMeshLikeceZdZUded<ded<ded<ded<ded <d ed <ded <dRdZdSdZdTdZdUdZdVdZdVdZ dWdZ dXdZ dXd Z dXd!Z dXd"ZdXd#Zd$d%dYd(ZdXd)ZdXd*ZdZd+Z d[d\d6Zd]d8Z d^d_d>Zd]d?Zd`d@Z dadbdAZdcdDZdcdEZedddGZdedHZdfdKZdXdLZ dMedN<dXdOZ!dXdPZ"dXdQZ#d,S)gTransformersLoggerstrnameintlevelzlogging.Logger | Noneparentbool propagatedisabledzlist[logging.Handler]handlersraiseExceptionsr returnNonecdSNselfrs BC:\Projects\DSCons\.venv\Lib\site-packages\transformers\_typing.pysetLevelzTransformersLogger.setLevel4cdSrrrs r! isEnabledForzTransformersLogger.isEnabledFor5r#r$cdSrrr s r!getEffectiveLevelz$TransformersLogger.getEffectiveLevel6r#r$suffixlogging.LoggercdSrr)r r*s r!getChildzTransformersLogger.getChild8r#r$hdlrlogging.HandlercdSrrr r.s r! addHandlerzTransformersLogger.addHandler:r#r$cdSrrr1s r! removeHandlerz TransformersLogger.removeHandler;r#r$cdSrrr(s r! hasHandlerszTransformersLogger.hasHandlers<r#r$msgobjectargskwargscdSrrr r7r9r:s r!debugzTransformersLogger.debug?r#r$cdSrrr<s r!infozTransformersLogger.info@r#r$cdSrrr<s r!warningzTransformersLogger.warningAr#r$cdSrrr<s r!warnzTransformersLogger.warnBr#r$cdSrrr<s r!errorzTransformersLogger.errorCr#r$T)exc_inforFr cdSrr)r r7rFr9r:s r! exceptionzTransformersLogger.exceptionDr#r$cdSrrr<s r!criticalzTransformersLogger.criticalEr#r$cdSrrr<s r!fatalzTransformersLogger.fatalFr#r$cdSrr)r rr7r9r:s r!logzTransformersLogger.logIr#r$Nfnlno)tuple[object, ...] | Mapping[str, object]func str | NoneextraMapping[str, object] | Nonesinfologging.LogRecordc dSrr) r rrrOrPr7r9rFrRrTrVs r! makeRecordzTransformersLogger.makeRecordLs  Cr$recordcdSrrr rZs r!handlezTransformersLogger.handleZr#r$F stack_info stacklevel tuple[str, int, str, str | None]cdSrr)r r_r`s r! findCallerzTransformersLogger.findCaller[s ,/3r$cdSrrr\s r! callHandlerszTransformersLogger.callHandlersar#r$cdSrrr(s r! getMessagezTransformersLogger.getMessagebr#r$cdSrr)r rr7r9rFrTr_r`s r!_logzTransformersLogger._logds sr$filtlogging.FiltercdSrrr rjs r! addFilterzTransformersLogger.addFilterpr#r$cdSrrrms r! removeFilterzTransformersLogger.removeFilterqr#r$list[logging.Filter]cdSrrr(s r!filterszTransformersLogger.filtersrs/2sr$cdSrrr\s r!filterzTransformersLogger.filterur#r$fmtlogging.FormattercdSrr)r rvs r! setFormatterzTransformersLogger.setFormatterxr#r$cdSrrr<s r! debugStackzTransformersLogger.debugStackyr#r$zMutableMapping[str, Any]__dict__cdSrrr<s r!warning_advicez!TransformersLogger.warning_advicer#r$cdSrrr<s r! warning_oncezTransformersLogger.warning_oncer#r$cdSrrr<s r! info_oncezTransformersLogger.info_oncer#r$)rr rr)rr rr)rr)r*rrr+)r.r/rrrr)r7r8r9r8r:r8rr) r7r8r9r8rFr r:r8rr) rr r7r8r9r8r:r8rr)NNN)rrrr rOrrPrr7r8r9rQrFr rRrSrTrUrVrSrrW)rZrWrr)Fr^)r_rr`rrra)rr)NNFr^)rr r7r8r9rQrFr rTrUr_rr`rrr)rjrkrr)rrq)rZrWrr)rvrwrr)$__name__ __module__ __qualname____annotations__r"r&r)r-r2r4r6r=r?rArCrErHrJrLrNrYr]rcrergrirnrppropertyrsruryr{r~rrrr$r!rr'sN IIIJJJ!!!!OOONNN####21115555++++::::<<<<????&&&&ONNNMMMMPPPPMMMMNNNNHLllllllQQQQNNNN[ZZZ -1       =<<<!///// CBBB$$$$!-1      ;:::==== 222X2<<<<@???SSSS '&&&XWWWUUUURRRRRRr$rceZdZUdZded<ded<ded<ded <ded <d ed <ded <ded<ded<ded<ded<d(dZd)dZd)dZd*dZd+dZ d+dZ d+dZ d,d!Z d,d"Z d-d$Zd)d%Zd*d&Zd'S).GenerativePreTrainedModeluwProtocol for the model interface that GenerationMixin expects. GenerationMixin is designed to be mixed into PreTrainedModel subclasses. This Protocol documents the attributes and methods the mixin relies on from its host class. It is *not* used at runtime — its purpose is to help the ``ty`` type checker resolve ``self.`` accesses inside the mixin. rconfigz torch.devicedevicez torch.dtypedtypermain_input_namebase_model_prefixr _is_stateful hf_quantizerencoderzdict[str, Any] hf_device_mapr_previous_max_cache_lengthgeneration_configrrcdSrr)r rs r! __getattr__z%GenerativePreTrainedModel.__getattr__r#r$r9r:cdSrrr r9r:s r!forwardz!GenerativePreTrainedModel.forwardr#r$cdSrrrs r!__call__z"GenerativePreTrainedModel.__call__r#r$cdSrrr(s r! can_generatez&GenerativePreTrainedModel.can_generater#r$cdSrrr(s r! get_encoderz%GenerativePreTrainedModel.get_encoderr#r$cdSrrr(s r!get_output_embeddingsz/GenerativePreTrainedModel.get_output_embeddingsr#r$cdSrrr(s r!get_input_embeddingsz.GenerativePreTrainedModel.get_input_embeddingsr#r$valuercdSrrr rs r!set_output_embeddingsz/GenerativePreTrainedModel.set_output_embeddingsr#r$cdSrrrs r!set_input_embeddingsz.GenerativePreTrainedModel.set_input_embeddingsr#r$compile_configcdSrr)r rs r!get_compiled_callz+GenerativePreTrainedModel.get_compiled_callr#r$cdSrrrs r!set_experts_implementationz4GenerativePreTrainedModel.set_experts_implementationr#r$cdSrrr(s r!_supports_logits_to_keepz2GenerativePreTrainedModel._supports_logits_to_keepr#r$N)rrrr)r9rr:rrrr)rr)rrrr)rrrr)rrr__doc__rrrrrrrrrrrrrrr$r!rrs,KKKLLL!!!!####0000<<<<====''''%%%%////....<<<<;;;;@@@@OOOO333333r$rceZdZUded<dS)StringValuedEnumLikerrN)rrrrrr$r!rrsJJJJJr$rc@eZdZUded<ded<ded<ded<ddZdS)PeftConfigLiker peft_typeris_prompt_learningzstr | PathLike[str] | Nonebase_model_name_or_pathinference_modesave_directorystr | PathLike[str]r:rrrc dSrr)r rr:s r!save_pretrainedzPeftConfigLike.save_pretrainedr#r$N)rrr:rrr)rrrrrrr$r!rrsN####7777^^^^^^r$rceZdZUdZded<dS)WhisperGenerationConfigLikezXProtocol for Whisper-specific generation config fields accessed in generation internals.rno_timestamps_token_idN)rrrrrrr$r!rrs$bbr$r)r __future__rloggingcollections.abcrrosrtypingrrr r torchrr rr BaseExceptiontupletyper8r r rrrrrrr$r!rs=<<""""""33333333::::::::::::LLL   D  678  \S\S\S\S\S\S\S\S~ 4 4 4 4 4 4 4 4F8_____X___     (     r$