o
    Zh<                     @   sf  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=G dydz dzedZ>G d{d| d|edZ?G d}d~ d~edZ@G dd dedZAG dd dedZBG dd dedZCG dd dedZDG dd dedZEG dd dedZFG dd dedZGG dd dedZHG dd dedZIG dd dedZJG dd dedZKG dd dedZLG dd dedZMG dd dedZNdd ZOdd ZPG dd dedZQdZRdd ZSG dd dedZTG dd dedZUG dd dedZVdd ZWdd ZXdd ZYdd ZZdd Z[dd Z\dd Z]dd Z^dd Z_G dd dedZ`dd Zadd ZbG ddÄ dedZcddń ZdG ddǄ dedZedS )   )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   R/var/www/auris/lib/python3.10/site-packages/transformers/utils/dummy_pt_objects.py__init__      zCache.__init__N__name__
__module____qualname__Z	_backendsr   r   r   r   r   r          r   )	metaclassc                   @   r   )CacheConfigr   c                 O   r   r   r	   r
   r   r   r   r      r   zCacheConfig.__init__Nr   r   r   r   r   r      r   r   c                   @   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   )HQQQuantizedCacher   c                 O   r   r   r	   r
   r   r   r   r   $   r   zHQQQuantizedCache.__init__Nr   r   r   r   r   r   !   r   r   c                   @   r   )HybridCacher   c                 O   r   r   r	   r
   r   r   r   r   +   r   zHybridCache.__init__Nr   r   r   r   r   r   (   r   r   c                   @   r   )
MambaCacher   c                 O   r   r   r	   r
   r   r   r   r   2   r   zMambaCache.__init__Nr   r   r   r   r   r   /   r   r   c                   @   r   )OffloadedCacher   c                 O   r   r   r	   r
   r   r   r   r   9   r   zOffloadedCache.__init__Nr   r   r   r   r   r   6   r   r   c                   @   r   )OffloadedStaticCacher   c                 O   r   r   r	   r
   r   r   r   r   @   r   zOffloadedStaticCache.__init__Nr   r   r   r   r   r   =   r   r   c                   @   r   )QuantizedCacher   c                 O   r   r   r	   r
   r   r   r   r   G   r   zQuantizedCache.__init__Nr   r   r   r   r   r    D   r   r    c                   @   r   )QuantizedCacheConfigr   c                 O   r   r   r	   r
   r   r   r   r   N   r   zQuantizedCacheConfig.__init__Nr   r   r   r   r   r!   K   r   r!   c                   @   r   )QuantoQuantizedCacher   c                 O   r   r   r	   r
   r   r   r   r   U   r   zQuantoQuantizedCache.__init__Nr   r   r   r   r   r"   R   r   r"   c                   @   r   )	SinkCacher   c                 O   r   r   r	   r
   r   r   r   r   \   r   zSinkCache.__init__Nr   r   r   r   r   r#   Y   r   r#   c                   @   r   )SlidingWindowCacher   c                 O   r   r   r	   r
   r   r   r   r   c   r   zSlidingWindowCache.__init__Nr   r   r   r   r   r$   `   r   r$   c                   @   r   )StaticCacher   c                 O   r   r   r	   r
   r   r   r   r   j   r   zStaticCache.__init__Nr   r   r   r   r   r%   g   r   r%   c                   @   r   )GlueDatasetr   c                 O   r   r   r	   r
   r   r   r   r   q   r   zGlueDataset.__init__Nr   r   r   r   r   r&   n   r   r&   c                   @   r   )GlueDataTrainingArgumentsr   c                 O   r   r   r	   r
   r   r   r   r   x   r   z"GlueDataTrainingArguments.__init__Nr   r   r   r   r   r'   u   r   r'   c                   @   r   )LineByLineTextDatasetr   c                 O   r   r   r	   r
   r   r   r   r      r   zLineByLineTextDataset.__init__Nr   r   r   r   r   r(   |   r   r(   c                   @   r   )LineByLineWithRefDatasetr   c                 O   r   r   r	   r
   r   r   r   r      r   z!LineByLineWithRefDataset.__init__Nr   r   r   r   r   r)      r   r)   c                   @   r   )LineByLineWithSOPTextDatasetr   c                 O   r   r   r	   r
   r   r   r   r      r   z%LineByLineWithSOPTextDataset.__init__Nr   r   r   r   r   r*      r   r*   c                   @   r   )SquadDatasetr   c                 O   r   r   r	   r
   r   r   r   r      r   zSquadDataset.__init__Nr   r   r   r   r   r+      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   )TextDatasetr   c                 O   r   r   r	   r
   r   r   r   r      r   zTextDataset.__init__Nr   r   r   r   r   r-      r   r-   c                   @   r   )$TextDatasetForNextSentencePredictionr   c                 O   r   r   r	   r
   r   r   r   r      r   z-TextDatasetForNextSentencePrediction.__init__Nr   r   r   r   r   r.      r   r.   c                   @   r   )#AlternatingCodebooksLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z,AlternatingCodebooksLogitsProcessor.__init__Nr   r   r   r   r   r/      r   r/   c                   @   r   )BayesianDetectorConfigr   c                 O   r   r   r	   r
   r   r   r   r      r   zBayesianDetectorConfig.__init__Nr   r   r   r   r   r0      r   r0   c                   @   r   )BayesianDetectorModelr   c                 O   r   r   r	   r
   r   r   r   r      r   zBayesianDetectorModel.__init__Nr   r   r   r   r   r1      r   r1   c                   @   r   )
BeamScorerr   c                 O   r   r   r	   r
   r   r   r   r      r   zBeamScorer.__init__Nr   r   r   r   r   r2      r   r2   c                   @   r   )BeamSearchScorerr   c                 O   r   r   r	   r
   r   r   r   r      r   zBeamSearchScorer.__init__Nr   r   r   r   r   r3      r   r3   c                   @   r   )%ClassifierFreeGuidanceLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z.ClassifierFreeGuidanceLogitsProcessor.__init__Nr   r   r   r   r   r4      r   r4   c                   @   r   )ConstrainedBeamSearchScorerr   c                 O   r   r   r	   r
   r   r   r   r      r   z$ConstrainedBeamSearchScorer.__init__Nr   r   r   r   r   r5      r   r5   c                   @   r   )
Constraintr   c                 O   r   r   r	   r
   r   r   r   r      r   zConstraint.__init__Nr   r   r   r   r   r6      r   r6   c                   @   r   )ConstraintListStater   c                 O   r   r   r	   r
   r   r   r   r      r   zConstraintListState.__init__Nr   r   r   r   r   r7      r   r7   c                   @   r   )DisjunctiveConstraintr   c                 O   r   r   r	   r
   r   r   r   r      r   zDisjunctiveConstraint.__init__Nr   r   r   r   r   r8      r   r8   c                   @   r   )#EncoderNoRepeatNGramLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z,EncoderNoRepeatNGramLogitsProcessor.__init__Nr   r   r   r   r   r9      r   r9   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   r?     r   r?   c                   @   r   )ForcedEOSTokenLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   '  r   z&ForcedEOSTokenLogitsProcessor.__init__Nr   r   r   r   r   r@   $  r   r@   c                   @   r   )GenerationMixinr   c                 O   r   r   r	   r
   r   r   r   r   .  r   zGenerationMixin.__init__Nr   r   r   r   r   rA   +  r   rA   c                   @   r   )HammingDiversityLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   5  r   z(HammingDiversityLogitsProcessor.__init__Nr   r   r   r   r   rB   2  r   rB   c                   @   r   )InfNanRemoveLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   <  r   z$InfNanRemoveLogitsProcessor.__init__Nr   r   r   r   r   rC   9  r   rC   c                   @   r   )LogitNormalizationr   c                 O   r   r   r	   r
   r   r   r   r   C  r   zLogitNormalization.__init__Nr   r   r   r   r   rD   @  r   rD   c                   @   r   )LogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   J  r   zLogitsProcessor.__init__Nr   r   r   r   r   rE   G  r   rE   c                   @   r   )LogitsProcessorListr   c                 O   r   r   r	   r
   r   r   r   r   Q  r   zLogitsProcessorList.__init__Nr   r   r   r   r   rF   N  r   rF   c                   @   r   )MaxLengthCriteriar   c                 O   r   r   r	   r
   r   r   r   r   X  r   zMaxLengthCriteria.__init__Nr   r   r   r   r   rG   U  r   rG   c                   @   r   )MaxTimeCriteriar   c                 O   r   r   r	   r
   r   r   r   r   _  r   zMaxTimeCriteria.__init__Nr   r   r   r   r   rH   \  r   rH   c                   @   r   )MinLengthLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   f  r   z!MinLengthLogitsProcessor.__init__Nr   r   r   r   r   rI   c  r   rI   c                   @   r   )!MinNewTokensLengthLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   m  r   z*MinNewTokensLengthLogitsProcessor.__init__Nr   r   r   r   r   rJ   j  r   rJ   c                   @   r   )MinPLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r   t  r   zMinPLogitsWarper.__init__Nr   r   r   r   r   rK   q  r   rK   c                   @   r   )NoBadWordsLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   {  r   z"NoBadWordsLogitsProcessor.__init__Nr   r   r   r   r   rL   x  r   rL   c                   @   r   )NoRepeatNGramLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z%NoRepeatNGramLogitsProcessor.__init__Nr   r   r   r   r   rM     r   rM   c                   @   r   )PhrasalConstraintr   c                 O   r   r   r	   r
   r   r   r   r     r   zPhrasalConstraint.__init__Nr   r   r   r   r   rN     r   rN   c                   @   r   ) PrefixConstrainedLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z)PrefixConstrainedLogitsProcessor.__init__Nr   r   r   r   r   rO     r   rO   c                   @   r   ) RepetitionPenaltyLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z)RepetitionPenaltyLogitsProcessor.__init__Nr   r   r   r   r   rP     r   rP   c                   @   r   )SequenceBiasLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z$SequenceBiasLogitsProcessor.__init__Nr   r   r   r   r   rQ     r   rQ   c                   @   r   )StoppingCriteriar   c                 O   r   r   r	   r
   r   r   r   r     r   zStoppingCriteria.__init__Nr   r   r   r   r   rR     r   rR   c                   @   r   )StoppingCriteriaListr   c                 O   r   r   r	   r
   r   r   r   r     r   zStoppingCriteriaList.__init__Nr   r   r   r   r   rS     r   rS   c                   @   r   )StopStringCriteriar   c                 O   r   r   r	   r
   r   r   r   r     r   zStopStringCriteria.__init__Nr   r   r   r   r   rT     r   rT   c                   @   r   )$SuppressTokensAtBeginLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z-SuppressTokensAtBeginLogitsProcessor.__init__Nr   r   r   r   r   rU     r   rU   c                   @   r   )SuppressTokensLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z&SuppressTokensLogitsProcessor.__init__Nr   r   r   r   r   rV     r   rV   c                   @   r   )SynthIDTextWatermarkDetectorr   c                 O   r   r   r	   r
   r   r   r   r     r   z%SynthIDTextWatermarkDetector.__init__Nr   r   r   r   r   rW     r   rW   c                   @   r   )SynthIDTextWatermarkingConfigr   c                 O   r   r   r	   r
   r   r   r   r     r   z&SynthIDTextWatermarkingConfig.__init__Nr   r   r   r   r   rX     r   rX   c                   @   r   )#SynthIDTextWatermarkLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z,SynthIDTextWatermarkLogitsProcessor.__init__Nr   r   r   r   r   rY     r   rY   c                   @   r   )TemperatureLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r     r   z TemperatureLogitsWarper.__init__Nr   r   r   r   r   rZ     r   rZ   c                   @   r   )TopKLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r     r   zTopKLogitsWarper.__init__Nr   r   r   r   r   r[     r   r[   c                   @   r   )TopPLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r     r   zTopPLogitsWarper.__init__Nr   r   r   r   r   r\     r   r\   c                   @   r   )TypicalLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r     r   zTypicalLogitsWarper.__init__Nr   r   r   r   r   r]     r   r]   c                   @   r   ).UnbatchedClassifierFreeGuidanceLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z7UnbatchedClassifierFreeGuidanceLogitsProcessor.__init__Nr   r   r   r   r   r^     r   r^   c                   @   r   )WatermarkDetectorr   c                 O   r   r   r	   r
   r   r   r   r      r   zWatermarkDetector.__init__Nr   r   r   r   r   r_     r   r_   c                   @   r   )WatermarkLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z!WatermarkLogitsProcessor.__init__Nr   r   r   r   r   r`     r   r`   c                   @   r   )WhisperTimeStampLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r     r   z(WhisperTimeStampLogitsProcessor.__init__Nr   r   r   r   r   ra     r   ra   c                   @   r   )$TorchExportableModuleWithStaticCacher   c                 O   r   r   r	   r
   r   r   r   r     r   z-TorchExportableModuleWithStaticCache.__init__Nr   r   r   r   r   rb     r   rb   c                  O      t tdg d S r   )r   convert_and_export_with_cacher   r   r   r   r   rd     r   rd   c                  O   rc   r   )r   model_addition_debugger_contextre   r   r   r   rf     r   rf   c                   @   r   )GradientCheckpointingLayerr   c                 O   r   r   r	   r
   r   r   r   r   $  r   z#GradientCheckpointingLayer.__init__Nr   r   r   r   r   rg   !  r   rg   Nc                  O   rc   r   )r   dynamic_rope_updatere   r   r   r   rh   +  r   rh   c                   @   r   )AttentionInterfacer   c                 O   r   r   r	   r
   r   r   r   r   2  r   zAttentionInterface.__init__Nr   r   r   r   r   ri   /  r   ri   c                   @   r   )PreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   9  r   zPreTrainedModel.__init__Nr   r   r   r   r   rj   6  r   rj   c                   @   r   )	Adafactorr   c                 O   r   r   r	   r
   r   r   r   r   @  r   zAdafactor.__init__Nr   r   r   r   r   rk   =  r   rk   c                  O   rc   r   )r   get_constant_schedulere   r   r   r   rl   D  r   rl   c                  O   rc   r   )r   !get_constant_schedule_with_warmupre   r   r   r   rm   H  r   rm   c                  O   rc   r   )r   get_cosine_schedule_with_warmupre   r   r   r   rn   L  r   rn   c                  O   rc   r   )r   2get_cosine_with_hard_restarts_schedule_with_warmupre   r   r   r   ro   P  r   ro   c                  O   rc   r   )r   get_inverse_sqrt_schedulere   r   r   r   rp   T  r   rp   c                  O   rc   r   )r   get_linear_schedule_with_warmupre   r   r   r   rq   X  r   rq   c                  O   rc   r   )r   )get_polynomial_decay_schedule_with_warmupre   r   r   r   rr   \  r   rr   c                  O   rc   r   )r   get_schedulerre   r   r   r   rs   `  r   rs   c                  O   rc   r   )r   get_wsd_schedulere   r   r   r   rt   d  r   rt   c                   @   r   )Conv1Dr   c                 O   r   r   r	   r
   r   r   r   r   k  r   zConv1D.__init__Nr   r   r   r   r   ru   h  r   ru   c                  O   rc   r   )r   apply_chunking_to_forwardre   r   r   r   rv   o  r   rv   c                  O   rc   r   )r   prune_layerre   r   r   r   rw   s  r   rw   c                   @   r   )Trainerr   c                 O   r   r   r	   r
   r   r   r   r   z  r   zTrainer.__init__Nr   r   r   r   r   rx   w  r   rx   c                  O   rc   r   )r   torch_distributed_zero_firstre   r   r   r   ry   ~  r   ry   c                   @   r   )Seq2SeqTrainerr   c                 O   r   r   r	   r
   r   r   r   r     r   zSeq2SeqTrainer.__init__Nr   r   r   r   r   rz     r   rz   )f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.   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   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rd   rf   rg   ZROPE_INIT_FUNCTIONSrh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   r   r   r   r   <module>   s   