o
    ei^2                     @   s>  d dl mZmZ G dd dedZG dd dedZG dd dedZG d	d
 d
edZG dd dedZG dd dedZG dd dedZ	G dd dedZ
G dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd  d edZG d!d" d"edZG d#d$ d$edZG d%d& d&edZG d'd( d(edZG d)d* d*edZG d+d, d,edZG d-d. d.edZG d/d0 d0edZG d1d2 d2edZG d3d4 d4edZG d5d6 d6edZG d7d8 d8edZG d9d: d:edZG d;d< d<edZG d=d> d>edZ G d?d@ d@edZ!G dAdB dBedZ"G dCdD dDedZ#G dEdF dFedZ$G dGdH dHedZ%G dIdJ dJedZ&G dKdL dLedZ'G dMdN dNedZ(G dOdP dPedZ)G dQdR dRedZ*G dSdT dTedZ+G dUdV dVedZ,G dWdX dXedZ-G dYdZ dZedZ.G d[d\ d\edZ/G d]d^ d^edZ0G d_d` d`edZ1G dadb dbedZ2G dcdd ddedZ3G dedf dfedZ4G dgdh dhedZ5G didj djedZ6G dkdl dledZ7G dmdn dnedZ8G dodp dpedZ9G dqdr dredZ:G dsdt dtedZ;G dudv dvedZ<G dwdx dxedZ=dydz Z>G d{d| d|edZ?d}d~ Z@G dd dedZAdZBdd ZCG dd dedZDG dd dedZEG dd dedZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNdd ZOG dd dedZPdd ZQG dd dedZRdd ZSG dd dedZTdS )   )DummyObjectrequires_backendsc                   @      e Zd ZdgZdd ZdS )Cachetorchc                 O      t | dg d S Nr   r   selfargskwargs r   a/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/transformers/utils/dummy_pt_objects.py__init__      zCache.__init__N__name__
__module____qualname__	_backendsr   r   r   r   r   r          r   )	metaclassc                   @   r   )DynamicCacher   c                 O   r   r   r	   r
   r   r   r   r      r   zDynamicCache.__init__Nr   r   r   r   r   r      r   r   c                   @   r   )EncoderDecoderCacher   c                 O   r   r   r	   r
   r   r   r   r      r   zEncoderDecoderCache.__init__Nr   r   r   r   r   r      r   r   c                   @   r   )QuantizedCacher   c                 O   r   r   r	   r
   r   r   r   r      r   zQuantizedCache.__init__Nr   r   r   r   r   r      r   r   c                   @   r   )StaticCacher   c                 O   r   r   r	   r
   r   r   r   r   $   r   zStaticCache.__init__Nr   r   r   r   r   r   !   r   r   c                   @   r   )GlueDatasetr   c                 O   r   r   r	   r
   r   r   r   r   +   r   zGlueDataset.__init__Nr   r   r   r   r   r   (   r   r   c                   @   r   )GlueDataTrainingArgumentsr   c                 O   r   r   r	   r
   r   r   r   r   2   r   z"GlueDataTrainingArguments.__init__Nr   r   r   r   r   r   /   r   r   c                   @   r   )SquadDatasetr   c                 O   r   r   r	   r
   r   r   r   r   9   r   zSquadDataset.__init__Nr   r   r   r   r   r   6   r   r   c                   @   r   )SquadDataTrainingArgumentsr   c                 O   r   r   r	   r
   r   r   r   r   @   r   z#SquadDataTrainingArguments.__init__Nr   r   r   r   r   r    =   r   r    c                   @   r   )#AlternatingCodebooksLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   G   r   z,AlternatingCodebooksLogitsProcessor.__init__Nr   r   r   r   r   r!   D   r   r!   c                   @   r   )BayesianDetectorConfigr   c                 O   r   r   r	   r
   r   r   r   r   N   r   zBayesianDetectorConfig.__init__Nr   r   r   r   r   r"   K   r   r"   c                   @   r   )BayesianDetectorModelr   c                 O   r   r   r	   r
   r   r   r   r   U   r   zBayesianDetectorModel.__init__Nr   r   r   r   r   r#   R   r   r#   c                   @   r   )
BeamScorerr   c                 O   r   r   r	   r
   r   r   r   r   \   r   zBeamScorer.__init__Nr   r   r   r   r   r$   Y   r   r$   c                   @   r   )%ClassifierFreeGuidanceLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   c   r   z.ClassifierFreeGuidanceLogitsProcessor.__init__Nr   r   r   r   r   r%   `   r   r%   c                   @   r   )ConstrainedBeamSearchScorerr   c                 O   r   r   r	   r
   r   r   r   r   j   r   z$ConstrainedBeamSearchScorer.__init__Nr   r   r   r   r   r&   g   r   r&   c                   @   r   )
Constraintr   c                 O   r   r   r	   r
   r   r   r   r   q   r   zConstraint.__init__Nr   r   r   r   r   r'   n   r   r'   c                   @   r   )ConstraintListStater   c                 O   r   r   r	   r
   r   r   r   r   x   r   zConstraintListState.__init__Nr   r   r   r   r   r(   u   r   r(   c                   @   r   )DisjunctiveConstraintr   c                 O   r   r   r	   r
   r   r   r   r      r   zDisjunctiveConstraint.__init__Nr   r   r   r   r   r)   |   r   r)   c                   @   r   )#EncoderNoRepeatNGramLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z,EncoderNoRepeatNGramLogitsProcessor.__init__Nr   r   r   r   r   r*      r   r*   c                   @   r   )'EncoderRepetitionPenaltyLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z0EncoderRepetitionPenaltyLogitsProcessor.__init__Nr   r   r   r   r   r+      r   r+   c                   @   r   )EosTokenCriteriar   c                 O   r   r   r	   r
   r   r   r   r      r   zEosTokenCriteria.__init__Nr   r   r   r   r   r,      r   r,   c                   @   r   )EpsilonLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r      r   zEpsilonLogitsWarper.__init__Nr   r   r   r   r   r-      r   r-   c                   @   r   )EtaLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r      r   zEtaLogitsWarper.__init__Nr   r   r   r   r   r.      r   r.   c                   @   r   )ExponentialDecayLengthPenaltyr   c                 O   r   r   r	   r
   r   r   r   r      r   z&ExponentialDecayLengthPenalty.__init__Nr   r   r   r   r   r/      r   r/   c                   @   r   )ForcedBOSTokenLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z&ForcedBOSTokenLogitsProcessor.__init__Nr   r   r   r   r   r0      r   r0   c                   @   r   )ForcedEOSTokenLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z&ForcedEOSTokenLogitsProcessor.__init__Nr   r   r   r   r   r1      r   r1   c                   @   r   )GenerationMixinr   c                 O   r   r   r	   r
   r   r   r   r      r   zGenerationMixin.__init__Nr   r   r   r   r   r2      r   r2   c                   @   r   )InfNanRemoveLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z$InfNanRemoveLogitsProcessor.__init__Nr   r   r   r   r   r3      r   r3   c                   @   r   )LogitNormalizationr   c                 O   r   r   r	   r
   r   r   r   r      r   zLogitNormalization.__init__Nr   r   r   r   r   r4      r   r4   c                   @   r   )LogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   zLogitsProcessor.__init__Nr   r   r   r   r   r5      r   r5   c                   @   r   )LogitsProcessorListr   c                 O   r   r   r	   r
   r   r   r   r      r   zLogitsProcessorList.__init__Nr   r   r   r   r   r6      r   r6   c                   @   r   )MaxLengthCriteriar   c                 O   r   r   r	   r
   r   r   r   r      r   zMaxLengthCriteria.__init__Nr   r   r   r   r   r7      r   r7   c                   @   r   )MaxTimeCriteriar   c                 O   r   r   r	   r
   r   r   r   r      r   zMaxTimeCriteria.__init__Nr   r   r   r   r   r8      r   r8   c                   @   r   )MinLengthLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z!MinLengthLogitsProcessor.__init__Nr   r   r   r   r   r9      r   r9   c                   @   r   )!MinNewTokensLengthLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z*MinNewTokensLengthLogitsProcessor.__init__Nr   r   r   r   r   r:      r   r:   c                   @   r   )MinPLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r      r   zMinPLogitsWarper.__init__Nr   r   r   r   r   r;      r   r;   c                   @   r   )NoBadWordsLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z"NoBadWordsLogitsProcessor.__init__Nr   r   r   r   r   r<     r   r<   c                   @   r   )NoRepeatNGramLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z%NoRepeatNGramLogitsProcessor.__init__Nr   r   r   r   r   r=     r   r=   c                   @   r   )PhrasalConstraintr   c                 O   r   r   r	   r
   r   r   r   r     r   zPhrasalConstraint.__init__Nr   r   r   r   r   r>     r   r>   c                   @   r   ) PrefixConstrainedLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z)PrefixConstrainedLogitsProcessor.__init__Nr   r   r   r   r   r?     r   r?   c                   @   r   ) RepetitionPenaltyLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z)RepetitionPenaltyLogitsProcessor.__init__Nr   r   r   r   r   r@     r   r@   c                   @   r   )SequenceBiasLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   '  r   z$SequenceBiasLogitsProcessor.__init__Nr   r   r   r   r   rA   $  r   rA   c                   @   r   )StoppingCriteriar   c                 O   r   r   r	   r
   r   r   r   r   .  r   zStoppingCriteria.__init__Nr   r   r   r   r   rB   +  r   rB   c                   @   r   )StoppingCriteriaListr   c                 O   r   r   r	   r
   r   r   r   r   5  r   zStoppingCriteriaList.__init__Nr   r   r   r   r   rC   2  r   rC   c                   @   r   )StopStringCriteriar   c                 O   r   r   r	   r
   r   r   r   r   <  r   zStopStringCriteria.__init__Nr   r   r   r   r   rD   9  r   rD   c                   @   r   )$SuppressTokensAtBeginLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   C  r   z-SuppressTokensAtBeginLogitsProcessor.__init__Nr   r   r   r   r   rE   @  r   rE   c                   @   r   )SuppressTokensLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   J  r   z&SuppressTokensLogitsProcessor.__init__Nr   r   r   r   r   rF   G  r   rF   c                   @   r   )SynthIDTextWatermarkDetectorr   c                 O   r   r   r	   r
   r   r   r   r   Q  r   z%SynthIDTextWatermarkDetector.__init__Nr   r   r   r   r   rG   N  r   rG   c                   @   r   )SynthIDTextWatermarkingConfigr   c                 O   r   r   r	   r
   r   r   r   r   X  r   z&SynthIDTextWatermarkingConfig.__init__Nr   r   r   r   r   rH   U  r   rH   c                   @   r   )#SynthIDTextWatermarkLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   _  r   z,SynthIDTextWatermarkLogitsProcessor.__init__Nr   r   r   r   r   rI   \  r   rI   c                   @   r   )TemperatureLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r   f  r   z TemperatureLogitsWarper.__init__Nr   r   r   r   r   rJ   c  r   rJ   c                   @   r   )TopKLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r   m  r   zTopKLogitsWarper.__init__Nr   r   r   r   r   rK   j  r   rK   c                   @   r   )TopPLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r   t  r   zTopPLogitsWarper.__init__Nr   r   r   r   r   rL   q  r   rL   c                   @   r   )TypicalLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r   {  r   zTypicalLogitsWarper.__init__Nr   r   r   r   r   rM   x  r   rM   c                   @   r   ).UnbatchedClassifierFreeGuidanceLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z7UnbatchedClassifierFreeGuidanceLogitsProcessor.__init__Nr   r   r   r   r   rN     r   rN   c                   @   r   )WatermarkDetectorr   c                 O   r   r   r	   r
   r   r   r   r     r   zWatermarkDetector.__init__Nr   r   r   r   r   rO     r   rO   c                   @   r   )WatermarkLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z!WatermarkLogitsProcessor.__init__Nr   r   r   r   r   rP     r   rP   c                   @   r   )WhisperTimeStampLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z(WhisperTimeStampLogitsProcessor.__init__Nr   r   r   r   r   rQ     r   rQ   c                   @   r   )$TorchExportableModuleWithStaticCacher   c                 O   r   r   r	   r
   r   r   r   r     r   z-TorchExportableModuleWithStaticCache.__init__Nr   r   r   r   r   rR     r   rR   c                  O      t tdg d S r   )r   convert_and_export_with_cacher   r   r   r   r   rT     r   rT   c                   @   r   )AttentionMaskInterfacer   c                 O   r   r   r	   r
   r   r   r   r     r   zAttentionMaskInterface.__init__Nr   r   r   r   r   rV     r   rV   c                  O   rS   r   )r   model_addition_debugger_contextrU   r   r   r   rW     r   rW   c                   @   r   )GradientCheckpointingLayerr   c                 O   r   r   r	   r
   r   r   r   r     r   z#GradientCheckpointingLayer.__init__Nr   r   r   r   r   rX     r   rX   Nc                  O   rS   r   )r   dynamic_rope_updaterU   r   r   r   rY     r   rY   c                   @   r   )AttentionInterfacer   c                 O   r   r   r	   r
   r   r   r   r     r   zAttentionInterface.__init__Nr   r   r   r   r   rZ     r   rZ   c                   @   r   )PreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zPreTrainedModel.__init__Nr   r   r   r   r   r[     r   r[   c                   @   r   )	Adafactorr   c                 O   r   r   r	   r
   r   r   r   r     r   zAdafactor.__init__Nr   r   r   r   r   r\     r   r\   c                  O   rS   r   )r   get_constant_schedulerU   r   r   r   r]     r   r]   c                  O   rS   r   )r   !get_constant_schedule_with_warmuprU   r   r   r   r^     r   r^   c                  O   rS   r   )r   get_cosine_schedule_with_warmuprU   r   r   r   r_     r   r_   c                  O   rS   r   )r   2get_cosine_with_hard_restarts_schedule_with_warmuprU   r   r   r   r`     r   r`   c                  O   rS   r   )r   get_inverse_sqrt_schedulerU   r   r   r   ra     r   ra   c                  O   rS   r   )r   get_linear_schedule_with_warmuprU   r   r   r   rb     r   rb   c                  O   rS   r   )r   )get_polynomial_decay_schedule_with_warmuprU   r   r   r   rc     r   rc   c                  O   rS   r   )r   get_schedulerrU   r   r   r   rd     r   rd   c                  O   rS   r   )r   get_wsd_schedulerU   r   r   r   re     r   re   c                   @   r   )Conv1Dr   c                 O   r   r   r	   r
   r   r   r   r     r   zConv1D.__init__Nr   r   r   r   r   rf     r   rf   c                  O   rS   r   )r   apply_chunking_to_forwardrU   r   r   r   rg     r   rg   c                   @   r   )Trainerr   c                 O   r   r   r	   r
   r   r   r   r     r   zTrainer.__init__Nr   r   r   r   r   rh     r   rh   c                  O   rS   r   )r   torch_distributed_zero_firstrU   r   r   r   ri   
  r   ri   c                   @   r   )Seq2SeqTrainerr   c                 O   r   r   r	   r
   r   r   r   r     r   zSeq2SeqTrainer.__init__Nr   r   r   r   r   rj     r   rj   )Uutilsr   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rT   rV   rW   rX   ROPE_INIT_FUNCTIONSrY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   r   r   r   r   <module>   s   