`i20>ddlmZddlmZmZmZmZmZgdgddZ es e gded<gd ed <gd ed <gd ed<gded<dged<gded<gded<n #e$rYnwxYw es e gded<gded<n #e$rYnwxYw es e gded<gded<n #e$rYnwxYwerddl m Z m Z m Z m Z mZmZddlmZmZmZmZ es e ddlmZmZmZmZdd lmZmZmZdd lmZm Z m!Z!m"Z"dd l#m$Z$dd!l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDdd"lEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMdd#lmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\dd$l]m^Z^m_Z_m`Z`maZambZbn #e$rYnwxYw es e ddlcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrdd%lsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~n #e$rYnwxYw es e dd&lmZmZmZmZmZmZmZmZmZmZmZmZmZmZdd'lmZmZmZmZd(S#e$rYd(SwxYwdd(lZeeed)ee*eje<d(S)+) TYPE_CHECKING)OptionalDependencyNotAvailable _LazyModuleis_flax_availableis_tf_availableis_torch_available)BaseWatermarkingConfig CompileConfigGenerationConfigGenerationModeSynthIDTextWatermarkingConfigWatermarkingConfig)AsyncTextIteratorStreamer BaseStreamerTextIteratorStreamer TextStreamer)configuration_utils streamers) ConstraintConstraintListStateDisjunctiveConstraintPhrasalConstraintbeam_constraints)BeamHypotheses BeamScorerConstrainedBeamSearchScorer beam_search)AssistedCandidateGeneratorCandidateGeneratorEarlyExitCandidateGeneratorPromptLookupCandidateGeneratorcandidate_generator)#AlternatingCodebooksLogitsProcessor%ClassifierFreeGuidanceLogitsProcessor#EncoderNoRepeatNGramLogitsProcessor'EncoderRepetitionPenaltyLogitsProcessorEpsilonLogitsWarperEtaLogitsWarperExponentialDecayLengthPenaltyForcedBOSTokenLogitsProcessorForcedEOSTokenLogitsProcessorInfNanRemoveLogitsProcessorLogitNormalizationLogitsProcessorLogitsProcessorListMinLengthLogitsProcessor!MinNewTokensLengthLogitsProcessorMinPLogitsWarperNoBadWordsLogitsProcessorNoRepeatNGramLogitsProcessor PrefixConstrainedLogitsProcessor RepetitionPenaltyLogitsProcessorSequenceBiasLogitsProcessorSuppressTokensLogitsProcessor$SuppressTokensAtBeginLogitsProcessor#SynthIDTextWatermarkLogitsProcessorTemperatureLogitsWarperTopKLogitsWarperTopPLogitsWarperTypicalLogitsWarper.UnbatchedClassifierFreeGuidanceLogitsProcessorWhisperTimeStampLogitsProcessorWatermarkLogitsProcessorlogits_process)MaxLengthCriteriaMaxTimeCriteriaConfidenceCriteriaEosTokenCriteriaStoppingCriteriaStoppingCriteriaListvalidate_stopping_criteriaStopStringCriteriastopping_criteriaContinuousMixincontinuous_batching)GenerationMixin GreedySearchEncoderDecoderOutputGreedySearchDecoderOnlyOutputSampleEncoderDecoderOutputSampleDecoderOnlyOutputBeamSearchEncoderDecoderOutputBeamSearchDecoderOnlyOutputBeamSampleEncoderDecoderOutputBeamSampleDecoderOnlyOutput%ContrastiveSearchEncoderDecoderOutput"ContrastiveSearchDecoderOnlyOutputGenerateBeamDecoderOnlyOutput GenerateBeamEncoderDecoderOutputGenerateDecoderOnlyOutputGenerateEncoderDecoderOutpututils)WatermarkDetectorWatermarkDetectorOutputBayesianDetectorModelBayesianDetectorConfigSynthIDTextWatermarkDetector watermarking)TFForcedBOSTokenLogitsProcessorTFForcedEOSTokenLogitsProcessorTFForceTokensLogitsProcessorTFLogitsProcessorTFLogitsProcessorListTFLogitsWarperTFMinLengthLogitsProcessorTFNoBadWordsLogitsProcessorTFNoRepeatNGramLogitsProcessor"TFRepetitionPenaltyLogitsProcessor&TFSuppressTokensAtBeginLogitsProcessorTFSuppressTokensLogitsProcessorTFTemperatureLogitsWarperTFTopKLogitsWarperTFTopPLogitsWarpertf_logits_process) TFGenerationMixinTFGreedySearchDecoderOnlyOutput"TFGreedySearchEncoderDecoderOutputTFSampleEncoderDecoderOutputTFSampleDecoderOnlyOutput TFBeamSearchEncoderDecoderOutputTFBeamSearchDecoderOnlyOutput TFBeamSampleEncoderDecoderOutputTFBeamSampleDecoderOnlyOutput'TFContrastiveSearchEncoderDecoderOutput$TFContrastiveSearchDecoderOnlyOutputtf_utils)!FlaxForcedBOSTokenLogitsProcessor!FlaxForcedEOSTokenLogitsProcessorFlaxForceTokensLogitsProcessorFlaxLogitsProcessorFlaxLogitsProcessorListFlaxLogitsWarperFlaxMinLengthLogitsProcessor(FlaxSuppressTokensAtBeginLogitsProcessor!FlaxSuppressTokensLogitsProcessorFlaxTemperatureLogitsWarperFlaxTopKLogitsWarperFlaxTopPLogitsWarper#FlaxWhisperTimeStampLogitsProcessor FlaxNoRepeatNGramLogitsProcessorflax_logits_process)FlaxGenerationMixinFlaxGreedySearchOutputFlaxSampleOutputFlaxBeamSearchOutput flax_utils)rM)r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r:r9r;r<r=r>r?r@rBrA)rFrGrDrErHrIrKrJ)rWrVrUrTrYrXrZr[r\r]rOrQrPrSrR)rbrarcr_r`) r}r|r{rzrr~rurvrwryrx)rrrrrrrrrrrrrr)rrrrN__file__) module_spec)typingrr^rrrrr _import_structurerr r r r rrrrrrrrrrrrrrrrr#rr r!r"rNrMrCr$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r:r9r;r<r=r>r?r@rBrArLrFrGrDrErHrIrKrJrWrVrUrTrYrXrZr[r\r]rOrQrPrSrRrdrbrarcr_r`rtrerfrgrhrirjrkrlrmrnrorprqrrrsrr}r|r{rzrr~rurvrwryrxrrrrrrrrrrrrrrrrrrrrsys__name__globals__spec__modulest/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/transformers/generation/__init__.pyrsS ! wwwwwwwwwwwwwwgff  \    /,,.../ ---() (((m$ 000+, + + +&'B . . .)* 0+,"""g")))n%%g&   D v# ?  /,,.../ ...)*" % % %j!!)&   D D    /,,.../ 000+, '''l##'&   D 4]rgfffffffffffQ !!## 30022 2 3 poooooooooooXXXXXXXXXX            988888 B                                   "               Q *     `#    30022 2 3                                   " ) *     Dt  "" 30022 2 3                                 tsssssssssssss' *     *JJJ'K''))J2GIZhpqqqCKskA,,A43A48BB$#B$(C CC7F;;GGHH! H!%I((I10I1