How to use requires method in Slash

Best Python code snippet using slash

dummy_pt_objects.py

Source:dummy_pt_objects.py Github

copy

Full Screen

1# This file is autogenerated by the command `make fix-copies`, do not edit.2from ..file_utils import requires_pytorch3class PyTorchBenchmark:4 def __init__(self, *args, **kwargs):5 requires_pytorch(self)6class PyTorchBenchmarkArguments:7 def __init__(self, *args, **kwargs):8 requires_pytorch(self)9class DataCollator:10 def __init__(self, *args, **kwargs):11 requires_pytorch(self)12class DataCollatorForLanguageModeling:13 def __init__(self, *args, **kwargs):14 requires_pytorch(self)15 @classmethod16 def from_pretrained(self, *args, **kwargs):17 requires_pytorch(self)18class DataCollatorForPermutationLanguageModeling:19 def __init__(self, *args, **kwargs):20 requires_pytorch(self)21 @classmethod22 def from_pretrained(self, *args, **kwargs):23 requires_pytorch(self)24class DataCollatorForSeq2Seq:25 def __init__(self, *args, **kwargs):26 requires_pytorch(self)27class DataCollatorForSOP:28 def __init__(self, *args, **kwargs):29 requires_pytorch(self)30class DataCollatorForTokenClassification:31 def __init__(self, *args, **kwargs):32 requires_pytorch(self)33 @classmethod34 def from_pretrained(self, *args, **kwargs):35 requires_pytorch(self)36class DataCollatorForWholeWordMask:37 def __init__(self, *args, **kwargs):38 requires_pytorch(self)39class DataCollatorWithPadding:40 def __init__(self, *args, **kwargs):41 requires_pytorch(self)42def default_data_collator(*args, **kwargs):43 requires_pytorch(default_data_collator)44class GlueDataset:45 def __init__(self, *args, **kwargs):46 requires_pytorch(self)47class GlueDataTrainingArguments:48 def __init__(self, *args, **kwargs):49 requires_pytorch(self)50class LineByLineTextDataset:51 def __init__(self, *args, **kwargs):52 requires_pytorch(self)53class LineByLineWithRefDataset:54 def __init__(self, *args, **kwargs):55 requires_pytorch(self)56class LineByLineWithSOPTextDataset:57 def __init__(self, *args, **kwargs):58 requires_pytorch(self)59class SquadDataset:60 def __init__(self, *args, **kwargs):61 requires_pytorch(self)62class SquadDataTrainingArguments:63 def __init__(self, *args, **kwargs):64 requires_pytorch(self)65class TextDataset:66 def __init__(self, *args, **kwargs):67 requires_pytorch(self)68class TextDatasetForNextSentencePrediction:69 def __init__(self, *args, **kwargs):70 requires_pytorch(self)71class BeamScorer:72 def __init__(self, *args, **kwargs):73 requires_pytorch(self)74class BeamSearchScorer:75 def __init__(self, *args, **kwargs):76 requires_pytorch(self)77class ForcedBOSTokenLogitsProcessor:78 def __init__(self, *args, **kwargs):79 requires_pytorch(self)80class ForcedEOSTokenLogitsProcessor:81 def __init__(self, *args, **kwargs):82 requires_pytorch(self)83class HammingDiversityLogitsProcessor:84 def __init__(self, *args, **kwargs):85 requires_pytorch(self)86class InfNanRemoveLogitsProcessor:87 def __init__(self, *args, **kwargs):88 requires_pytorch(self)89class LogitsProcessor:90 def __init__(self, *args, **kwargs):91 requires_pytorch(self)92class LogitsProcessorList:93 def __init__(self, *args, **kwargs):94 requires_pytorch(self)95class LogitsWarper:96 def __init__(self, *args, **kwargs):97 requires_pytorch(self)98class MinLengthLogitsProcessor:99 def __init__(self, *args, **kwargs):100 requires_pytorch(self)101class NoBadWordsLogitsProcessor:102 def __init__(self, *args, **kwargs):103 requires_pytorch(self)104class NoRepeatNGramLogitsProcessor:105 def __init__(self, *args, **kwargs):106 requires_pytorch(self)107class PrefixConstrainedLogitsProcessor:108 def __init__(self, *args, **kwargs):109 requires_pytorch(self)110class RepetitionPenaltyLogitsProcessor:111 def __init__(self, *args, **kwargs):112 requires_pytorch(self)113class TemperatureLogitsWarper:114 def __init__(self, *args, **kwargs):115 requires_pytorch(self)116class TopKLogitsWarper:117 def __init__(self, *args, **kwargs):118 requires_pytorch(self)119class TopPLogitsWarper:120 def __init__(self, *args, **kwargs):121 requires_pytorch(self)122class MaxLengthCriteria:123 def __init__(self, *args, **kwargs):124 requires_pytorch(self)125class MaxTimeCriteria:126 def __init__(self, *args, **kwargs):127 requires_pytorch(self)128class StoppingCriteria:129 def __init__(self, *args, **kwargs):130 requires_pytorch(self)131class StoppingCriteriaList:132 def __init__(self, *args, **kwargs):133 requires_pytorch(self)134def top_k_top_p_filtering(*args, **kwargs):135 requires_pytorch(top_k_top_p_filtering)136class Conv1D:137 def __init__(self, *args, **kwargs):138 requires_pytorch(self)139class PreTrainedModel:140 def __init__(self, *args, **kwargs):141 requires_pytorch(self)142 @classmethod143 def from_pretrained(self, *args, **kwargs):144 requires_pytorch(self)145def apply_chunking_to_forward(*args, **kwargs):146 requires_pytorch(apply_chunking_to_forward)147def prune_layer(*args, **kwargs):148 requires_pytorch(prune_layer)149ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None150class AlbertForMaskedLM:151 def __init__(self, *args, **kwargs):152 requires_pytorch(self)153 @classmethod154 def from_pretrained(self, *args, **kwargs):155 requires_pytorch(self)156class AlbertForMultipleChoice:157 def __init__(self, *args, **kwargs):158 requires_pytorch(self)159 @classmethod160 def from_pretrained(self, *args, **kwargs):161 requires_pytorch(self)162class AlbertForPreTraining:163 def __init__(self, *args, **kwargs):164 requires_pytorch(self)165class AlbertForQuestionAnswering:166 def __init__(self, *args, **kwargs):167 requires_pytorch(self)168 @classmethod169 def from_pretrained(self, *args, **kwargs):170 requires_pytorch(self)171class AlbertForSequenceClassification:172 def __init__(self, *args, **kwargs):173 requires_pytorch(self)174 @classmethod175 def from_pretrained(self, *args, **kwargs):176 requires_pytorch(self)177class AlbertForTokenClassification:178 def __init__(self, *args, **kwargs):179 requires_pytorch(self)180 @classmethod181 def from_pretrained(self, *args, **kwargs):182 requires_pytorch(self)183class AlbertModel:184 def __init__(self, *args, **kwargs):185 requires_pytorch(self)186 @classmethod187 def from_pretrained(self, *args, **kwargs):188 requires_pytorch(self)189class AlbertPreTrainedModel:190 def __init__(self, *args, **kwargs):191 requires_pytorch(self)192 @classmethod193 def from_pretrained(self, *args, **kwargs):194 requires_pytorch(self)195def load_tf_weights_in_albert(*args, **kwargs):196 requires_pytorch(load_tf_weights_in_albert)197MODEL_FOR_CAUSAL_LM_MAPPING = None198MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING = None199MODEL_FOR_MASKED_LM_MAPPING = None200MODEL_FOR_MULTIPLE_CHOICE_MAPPING = None201MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING = None202MODEL_FOR_PRETRAINING_MAPPING = None203MODEL_FOR_QUESTION_ANSWERING_MAPPING = None204MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING = None205MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = None206MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING = None207MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING = None208MODEL_MAPPING = None209MODEL_WITH_LM_HEAD_MAPPING = None210class AutoModel:211 def __init__(self, *args, **kwargs):212 requires_pytorch(self)213 @classmethod214 def from_pretrained(self, *args, **kwargs):215 requires_pytorch(self)216class AutoModelForCausalLM:217 def __init__(self, *args, **kwargs):218 requires_pytorch(self)219 @classmethod220 def from_pretrained(self, *args, **kwargs):221 requires_pytorch(self)222class AutoModelForMaskedLM:223 def __init__(self, *args, **kwargs):224 requires_pytorch(self)225 @classmethod226 def from_pretrained(self, *args, **kwargs):227 requires_pytorch(self)228class AutoModelForMultipleChoice:229 def __init__(self, *args, **kwargs):230 requires_pytorch(self)231 @classmethod232 def from_pretrained(self, *args, **kwargs):233 requires_pytorch(self)234class AutoModelForNextSentencePrediction:235 def __init__(self, *args, **kwargs):236 requires_pytorch(self)237 @classmethod238 def from_pretrained(self, *args, **kwargs):239 requires_pytorch(self)240class AutoModelForPreTraining:241 def __init__(self, *args, **kwargs):242 requires_pytorch(self)243 @classmethod244 def from_pretrained(self, *args, **kwargs):245 requires_pytorch(self)246class AutoModelForQuestionAnswering:247 def __init__(self, *args, **kwargs):248 requires_pytorch(self)249 @classmethod250 def from_pretrained(self, *args, **kwargs):251 requires_pytorch(self)252class AutoModelForSeq2SeqLM:253 def __init__(self, *args, **kwargs):254 requires_pytorch(self)255 @classmethod256 def from_pretrained(self, *args, **kwargs):257 requires_pytorch(self)258class AutoModelForSequenceClassification:259 def __init__(self, *args, **kwargs):260 requires_pytorch(self)261 @classmethod262 def from_pretrained(self, *args, **kwargs):263 requires_pytorch(self)264class AutoModelForTableQuestionAnswering:265 def __init__(self, *args, **kwargs):266 requires_pytorch(self)267 @classmethod268 def from_pretrained(self, *args, **kwargs):269 requires_pytorch(self)270class AutoModelForTokenClassification:271 def __init__(self, *args, **kwargs):272 requires_pytorch(self)273 @classmethod274 def from_pretrained(self, *args, **kwargs):275 requires_pytorch(self)276class AutoModelWithLMHead:277 def __init__(self, *args, **kwargs):278 requires_pytorch(self)279 @classmethod280 def from_pretrained(self, *args, **kwargs):281 requires_pytorch(self)282BART_PRETRAINED_MODEL_ARCHIVE_LIST = None283class BartForCausalLM:284 def __init__(self, *args, **kwargs):285 requires_pytorch(self)286class BartForConditionalGeneration:287 def __init__(self, *args, **kwargs):288 requires_pytorch(self)289 @classmethod290 def from_pretrained(self, *args, **kwargs):291 requires_pytorch(self)292class BartForQuestionAnswering:293 def __init__(self, *args, **kwargs):294 requires_pytorch(self)295 @classmethod296 def from_pretrained(self, *args, **kwargs):297 requires_pytorch(self)298class BartForSequenceClassification:299 def __init__(self, *args, **kwargs):300 requires_pytorch(self)301 @classmethod302 def from_pretrained(self, *args, **kwargs):303 requires_pytorch(self)304class BartModel:305 def __init__(self, *args, **kwargs):306 requires_pytorch(self)307 @classmethod308 def from_pretrained(self, *args, **kwargs):309 requires_pytorch(self)310class BartPretrainedModel:311 def __init__(self, *args, **kwargs):312 requires_pytorch(self)313 @classmethod314 def from_pretrained(self, *args, **kwargs):315 requires_pytorch(self)316class PretrainedBartModel:317 def __init__(self, *args, **kwargs):318 requires_pytorch(self)319 @classmethod320 def from_pretrained(self, *args, **kwargs):321 requires_pytorch(self)322BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None323class BertForMaskedLM:324 def __init__(self, *args, **kwargs):325 requires_pytorch(self)326 @classmethod327 def from_pretrained(self, *args, **kwargs):328 requires_pytorch(self)329class BertForMultipleChoice:330 def __init__(self, *args, **kwargs):331 requires_pytorch(self)332 @classmethod333 def from_pretrained(self, *args, **kwargs):334 requires_pytorch(self)335class BertForNextSentencePrediction:336 def __init__(self, *args, **kwargs):337 requires_pytorch(self)338class BertForPreTraining:339 def __init__(self, *args, **kwargs):340 requires_pytorch(self)341class BertForQuestionAnswering:342 def __init__(self, *args, **kwargs):343 requires_pytorch(self)344 @classmethod345 def from_pretrained(self, *args, **kwargs):346 requires_pytorch(self)347class BertForSequenceClassification:348 def __init__(self, *args, **kwargs):349 requires_pytorch(self)350 @classmethod351 def from_pretrained(self, *args, **kwargs):352 requires_pytorch(self)353class BertForTokenClassification:354 def __init__(self, *args, **kwargs):355 requires_pytorch(self)356 @classmethod357 def from_pretrained(self, *args, **kwargs):358 requires_pytorch(self)359class BertLayer:360 def __init__(self, *args, **kwargs):361 requires_pytorch(self)362class BertLMHeadModel:363 def __init__(self, *args, **kwargs):364 requires_pytorch(self)365 @classmethod366 def from_pretrained(self, *args, **kwargs):367 requires_pytorch(self)368class BertModel:369 def __init__(self, *args, **kwargs):370 requires_pytorch(self)371 @classmethod372 def from_pretrained(self, *args, **kwargs):373 requires_pytorch(self)374class BertPreTrainedModel:375 def __init__(self, *args, **kwargs):376 requires_pytorch(self)377 @classmethod378 def from_pretrained(self, *args, **kwargs):379 requires_pytorch(self)380def load_tf_weights_in_bert(*args, **kwargs):381 requires_pytorch(load_tf_weights_in_bert)382class BertGenerationDecoder:383 def __init__(self, *args, **kwargs):384 requires_pytorch(self)385class BertGenerationEncoder:386 def __init__(self, *args, **kwargs):387 requires_pytorch(self)388def load_tf_weights_in_bert_generation(*args, **kwargs):389 requires_pytorch(load_tf_weights_in_bert_generation)390BIG_BIRD_PRETRAINED_MODEL_ARCHIVE_LIST = None391class BigBirdForCausalLM:392 def __init__(self, *args, **kwargs):393 requires_pytorch(self)394class BigBirdForMaskedLM:395 def __init__(self, *args, **kwargs):396 requires_pytorch(self)397 @classmethod398 def from_pretrained(self, *args, **kwargs):399 requires_pytorch(self)400class BigBirdForMultipleChoice:401 def __init__(self, *args, **kwargs):402 requires_pytorch(self)403 @classmethod404 def from_pretrained(self, *args, **kwargs):405 requires_pytorch(self)406class BigBirdForPreTraining:407 def __init__(self, *args, **kwargs):408 requires_pytorch(self)409class BigBirdForQuestionAnswering:410 def __init__(self, *args, **kwargs):411 requires_pytorch(self)412 @classmethod413 def from_pretrained(self, *args, **kwargs):414 requires_pytorch(self)415class BigBirdForSequenceClassification:416 def __init__(self, *args, **kwargs):417 requires_pytorch(self)418 @classmethod419 def from_pretrained(self, *args, **kwargs):420 requires_pytorch(self)421class BigBirdForTokenClassification:422 def __init__(self, *args, **kwargs):423 requires_pytorch(self)424 @classmethod425 def from_pretrained(self, *args, **kwargs):426 requires_pytorch(self)427class BigBirdLayer:428 def __init__(self, *args, **kwargs):429 requires_pytorch(self)430class BigBirdModel:431 def __init__(self, *args, **kwargs):432 requires_pytorch(self)433 @classmethod434 def from_pretrained(self, *args, **kwargs):435 requires_pytorch(self)436class BigBirdPreTrainedModel:437 def __init__(self, *args, **kwargs):438 requires_pytorch(self)439 @classmethod440 def from_pretrained(self, *args, **kwargs):441 requires_pytorch(self)442def load_tf_weights_in_big_bird(*args, **kwargs):443 requires_pytorch(load_tf_weights_in_big_bird)444BLENDERBOT_PRETRAINED_MODEL_ARCHIVE_LIST = None445class BlenderbotForCausalLM:446 def __init__(self, *args, **kwargs):447 requires_pytorch(self)448class BlenderbotForConditionalGeneration:449 def __init__(self, *args, **kwargs):450 requires_pytorch(self)451 @classmethod452 def from_pretrained(self, *args, **kwargs):453 requires_pytorch(self)454class BlenderbotModel:455 def __init__(self, *args, **kwargs):456 requires_pytorch(self)457 @classmethod458 def from_pretrained(self, *args, **kwargs):459 requires_pytorch(self)460BLENDERBOT_SMALL_PRETRAINED_MODEL_ARCHIVE_LIST = None461class BlenderbotSmallForCausalLM:462 def __init__(self, *args, **kwargs):463 requires_pytorch(self)464class BlenderbotSmallForConditionalGeneration:465 def __init__(self, *args, **kwargs):466 requires_pytorch(self)467 @classmethod468 def from_pretrained(self, *args, **kwargs):469 requires_pytorch(self)470class BlenderbotSmallModel:471 def __init__(self, *args, **kwargs):472 requires_pytorch(self)473 @classmethod474 def from_pretrained(self, *args, **kwargs):475 requires_pytorch(self)476CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None477class CamembertForCausalLM:478 def __init__(self, *args, **kwargs):479 requires_pytorch(self)480class CamembertForMaskedLM:481 def __init__(self, *args, **kwargs):482 requires_pytorch(self)483 @classmethod484 def from_pretrained(self, *args, **kwargs):485 requires_pytorch(self)486class CamembertForMultipleChoice:487 def __init__(self, *args, **kwargs):488 requires_pytorch(self)489 @classmethod490 def from_pretrained(self, *args, **kwargs):491 requires_pytorch(self)492class CamembertForQuestionAnswering:493 def __init__(self, *args, **kwargs):494 requires_pytorch(self)495 @classmethod496 def from_pretrained(self, *args, **kwargs):497 requires_pytorch(self)498class CamembertForSequenceClassification:499 def __init__(self, *args, **kwargs):500 requires_pytorch(self)501 @classmethod502 def from_pretrained(self, *args, **kwargs):503 requires_pytorch(self)504class CamembertForTokenClassification:505 def __init__(self, *args, **kwargs):506 requires_pytorch(self)507 @classmethod508 def from_pretrained(self, *args, **kwargs):509 requires_pytorch(self)510class CamembertModel:511 def __init__(self, *args, **kwargs):512 requires_pytorch(self)513 @classmethod514 def from_pretrained(self, *args, **kwargs):515 requires_pytorch(self)516CONVBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None517class ConvBertForMaskedLM:518 def __init__(self, *args, **kwargs):519 requires_pytorch(self)520 @classmethod521 def from_pretrained(self, *args, **kwargs):522 requires_pytorch(self)523class ConvBertForMultipleChoice:524 def __init__(self, *args, **kwargs):525 requires_pytorch(self)526 @classmethod527 def from_pretrained(self, *args, **kwargs):528 requires_pytorch(self)529class ConvBertForQuestionAnswering:530 def __init__(self, *args, **kwargs):531 requires_pytorch(self)532 @classmethod533 def from_pretrained(self, *args, **kwargs):534 requires_pytorch(self)535class ConvBertForSequenceClassification:536 def __init__(self, *args, **kwargs):537 requires_pytorch(self)538 @classmethod539 def from_pretrained(self, *args, **kwargs):540 requires_pytorch(self)541class ConvBertForTokenClassification:542 def __init__(self, *args, **kwargs):543 requires_pytorch(self)544 @classmethod545 def from_pretrained(self, *args, **kwargs):546 requires_pytorch(self)547class ConvBertLayer:548 def __init__(self, *args, **kwargs):549 requires_pytorch(self)550class ConvBertModel:551 def __init__(self, *args, **kwargs):552 requires_pytorch(self)553 @classmethod554 def from_pretrained(self, *args, **kwargs):555 requires_pytorch(self)556class ConvBertPreTrainedModel:557 def __init__(self, *args, **kwargs):558 requires_pytorch(self)559 @classmethod560 def from_pretrained(self, *args, **kwargs):561 requires_pytorch(self)562def load_tf_weights_in_convbert(*args, **kwargs):563 requires_pytorch(load_tf_weights_in_convbert)564CTRL_PRETRAINED_MODEL_ARCHIVE_LIST = None565class CTRLForSequenceClassification:566 def __init__(self, *args, **kwargs):567 requires_pytorch(self)568 @classmethod569 def from_pretrained(self, *args, **kwargs):570 requires_pytorch(self)571class CTRLLMHeadModel:572 def __init__(self, *args, **kwargs):573 requires_pytorch(self)574 @classmethod575 def from_pretrained(self, *args, **kwargs):576 requires_pytorch(self)577class CTRLModel:578 def __init__(self, *args, **kwargs):579 requires_pytorch(self)580 @classmethod581 def from_pretrained(self, *args, **kwargs):582 requires_pytorch(self)583class CTRLPreTrainedModel:584 def __init__(self, *args, **kwargs):585 requires_pytorch(self)586 @classmethod587 def from_pretrained(self, *args, **kwargs):588 requires_pytorch(self)589DEBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None590class DebertaForMaskedLM:591 def __init__(self, *args, **kwargs):592 requires_pytorch(self)593 @classmethod594 def from_pretrained(self, *args, **kwargs):595 requires_pytorch(self)596class DebertaForQuestionAnswering:597 def __init__(self, *args, **kwargs):598 requires_pytorch(self)599 @classmethod600 def from_pretrained(self, *args, **kwargs):601 requires_pytorch(self)602class DebertaForSequenceClassification:603 def __init__(self, *args, **kwargs):604 requires_pytorch(self)605 @classmethod606 def from_pretrained(self, *args, **kwargs):607 requires_pytorch(self)608class DebertaForTokenClassification:609 def __init__(self, *args, **kwargs):610 requires_pytorch(self)611 @classmethod612 def from_pretrained(self, *args, **kwargs):613 requires_pytorch(self)614class DebertaModel:615 def __init__(self, *args, **kwargs):616 requires_pytorch(self)617 @classmethod618 def from_pretrained(self, *args, **kwargs):619 requires_pytorch(self)620class DebertaPreTrainedModel:621 def __init__(self, *args, **kwargs):622 requires_pytorch(self)623 @classmethod624 def from_pretrained(self, *args, **kwargs):625 requires_pytorch(self)626DEBERTA_V2_PRETRAINED_MODEL_ARCHIVE_LIST = None627class DebertaV2ForMaskedLM:628 def __init__(self, *args, **kwargs):629 requires_pytorch(self)630 @classmethod631 def from_pretrained(self, *args, **kwargs):632 requires_pytorch(self)633class DebertaV2ForQuestionAnswering:634 def __init__(self, *args, **kwargs):635 requires_pytorch(self)636 @classmethod637 def from_pretrained(self, *args, **kwargs):638 requires_pytorch(self)639class DebertaV2ForSequenceClassification:640 def __init__(self, *args, **kwargs):641 requires_pytorch(self)642 @classmethod643 def from_pretrained(self, *args, **kwargs):644 requires_pytorch(self)645class DebertaV2ForTokenClassification:646 def __init__(self, *args, **kwargs):647 requires_pytorch(self)648 @classmethod649 def from_pretrained(self, *args, **kwargs):650 requires_pytorch(self)651class DebertaV2Model:652 def __init__(self, *args, **kwargs):653 requires_pytorch(self)654 @classmethod655 def from_pretrained(self, *args, **kwargs):656 requires_pytorch(self)657class DebertaV2PreTrainedModel:658 def __init__(self, *args, **kwargs):659 requires_pytorch(self)660 @classmethod661 def from_pretrained(self, *args, **kwargs):662 requires_pytorch(self)663DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None664class DistilBertForMaskedLM:665 def __init__(self, *args, **kwargs):666 requires_pytorch(self)667 @classmethod668 def from_pretrained(self, *args, **kwargs):669 requires_pytorch(self)670class DistilBertForMultipleChoice:671 def __init__(self, *args, **kwargs):672 requires_pytorch(self)673 @classmethod674 def from_pretrained(self, *args, **kwargs):675 requires_pytorch(self)676class DistilBertForQuestionAnswering:677 def __init__(self, *args, **kwargs):678 requires_pytorch(self)679 @classmethod680 def from_pretrained(self, *args, **kwargs):681 requires_pytorch(self)682class DistilBertForSequenceClassification:683 def __init__(self, *args, **kwargs):684 requires_pytorch(self)685 @classmethod686 def from_pretrained(self, *args, **kwargs):687 requires_pytorch(self)688class DistilBertForTokenClassification:689 def __init__(self, *args, **kwargs):690 requires_pytorch(self)691 @classmethod692 def from_pretrained(self, *args, **kwargs):693 requires_pytorch(self)694class DistilBertModel:695 def __init__(self, *args, **kwargs):696 requires_pytorch(self)697 @classmethod698 def from_pretrained(self, *args, **kwargs):699 requires_pytorch(self)700class DistilBertPreTrainedModel:701 def __init__(self, *args, **kwargs):702 requires_pytorch(self)703 @classmethod704 def from_pretrained(self, *args, **kwargs):705 requires_pytorch(self)706DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None707DPR_QUESTION_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None708DPR_READER_PRETRAINED_MODEL_ARCHIVE_LIST = None709class DPRContextEncoder:710 def __init__(self, *args, **kwargs):711 requires_pytorch(self)712class DPRPretrainedContextEncoder:713 def __init__(self, *args, **kwargs):714 requires_pytorch(self)715class DPRPretrainedQuestionEncoder:716 def __init__(self, *args, **kwargs):717 requires_pytorch(self)718class DPRPretrainedReader:719 def __init__(self, *args, **kwargs):720 requires_pytorch(self)721class DPRQuestionEncoder:722 def __init__(self, *args, **kwargs):723 requires_pytorch(self)724class DPRReader:725 def __init__(self, *args, **kwargs):726 requires_pytorch(self)727ELECTRA_PRETRAINED_MODEL_ARCHIVE_LIST = None728class ElectraForMaskedLM:729 def __init__(self, *args, **kwargs):730 requires_pytorch(self)731 @classmethod732 def from_pretrained(self, *args, **kwargs):733 requires_pytorch(self)734class ElectraForMultipleChoice:735 def __init__(self, *args, **kwargs):736 requires_pytorch(self)737 @classmethod738 def from_pretrained(self, *args, **kwargs):739 requires_pytorch(self)740class ElectraForPreTraining:741 def __init__(self, *args, **kwargs):742 requires_pytorch(self)743class ElectraForQuestionAnswering:744 def __init__(self, *args, **kwargs):745 requires_pytorch(self)746 @classmethod747 def from_pretrained(self, *args, **kwargs):748 requires_pytorch(self)749class ElectraForSequenceClassification:750 def __init__(self, *args, **kwargs):751 requires_pytorch(self)752 @classmethod753 def from_pretrained(self, *args, **kwargs):754 requires_pytorch(self)755class ElectraForTokenClassification:756 def __init__(self, *args, **kwargs):757 requires_pytorch(self)758 @classmethod759 def from_pretrained(self, *args, **kwargs):760 requires_pytorch(self)761class ElectraModel:762 def __init__(self, *args, **kwargs):763 requires_pytorch(self)764 @classmethod765 def from_pretrained(self, *args, **kwargs):766 requires_pytorch(self)767class ElectraPreTrainedModel:768 def __init__(self, *args, **kwargs):769 requires_pytorch(self)770 @classmethod771 def from_pretrained(self, *args, **kwargs):772 requires_pytorch(self)773def load_tf_weights_in_electra(*args, **kwargs):774 requires_pytorch(load_tf_weights_in_electra)775class EncoderDecoderModel:776 def __init__(self, *args, **kwargs):777 requires_pytorch(self)778 @classmethod779 def from_pretrained(self, *args, **kwargs):780 requires_pytorch(self)781FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None782class FlaubertForMultipleChoice:783 def __init__(self, *args, **kwargs):784 requires_pytorch(self)785 @classmethod786 def from_pretrained(self, *args, **kwargs):787 requires_pytorch(self)788class FlaubertForQuestionAnswering:789 def __init__(self, *args, **kwargs):790 requires_pytorch(self)791 @classmethod792 def from_pretrained(self, *args, **kwargs):793 requires_pytorch(self)794class FlaubertForQuestionAnsweringSimple:795 def __init__(self, *args, **kwargs):796 requires_pytorch(self)797 @classmethod798 def from_pretrained(self, *args, **kwargs):799 requires_pytorch(self)800class FlaubertForSequenceClassification:801 def __init__(self, *args, **kwargs):802 requires_pytorch(self)803 @classmethod804 def from_pretrained(self, *args, **kwargs):805 requires_pytorch(self)806class FlaubertForTokenClassification:807 def __init__(self, *args, **kwargs):808 requires_pytorch(self)809 @classmethod810 def from_pretrained(self, *args, **kwargs):811 requires_pytorch(self)812class FlaubertModel:813 def __init__(self, *args, **kwargs):814 requires_pytorch(self)815 @classmethod816 def from_pretrained(self, *args, **kwargs):817 requires_pytorch(self)818class FlaubertWithLMHeadModel:819 def __init__(self, *args, **kwargs):820 requires_pytorch(self)821 @classmethod822 def from_pretrained(self, *args, **kwargs):823 requires_pytorch(self)824class FSMTForConditionalGeneration:825 def __init__(self, *args, **kwargs):826 requires_pytorch(self)827 @classmethod828 def from_pretrained(self, *args, **kwargs):829 requires_pytorch(self)830class FSMTModel:831 def __init__(self, *args, **kwargs):832 requires_pytorch(self)833 @classmethod834 def from_pretrained(self, *args, **kwargs):835 requires_pytorch(self)836class PretrainedFSMTModel:837 def __init__(self, *args, **kwargs):838 requires_pytorch(self)839 @classmethod840 def from_pretrained(self, *args, **kwargs):841 requires_pytorch(self)842FUNNEL_PRETRAINED_MODEL_ARCHIVE_LIST = None843class FunnelBaseModel:844 def __init__(self, *args, **kwargs):845 requires_pytorch(self)846 @classmethod847 def from_pretrained(self, *args, **kwargs):848 requires_pytorch(self)849class FunnelForMaskedLM:850 def __init__(self, *args, **kwargs):851 requires_pytorch(self)852 @classmethod853 def from_pretrained(self, *args, **kwargs):854 requires_pytorch(self)855class FunnelForMultipleChoice:856 def __init__(self, *args, **kwargs):857 requires_pytorch(self)858 @classmethod859 def from_pretrained(self, *args, **kwargs):860 requires_pytorch(self)861class FunnelForPreTraining:862 def __init__(self, *args, **kwargs):863 requires_pytorch(self)864class FunnelForQuestionAnswering:865 def __init__(self, *args, **kwargs):866 requires_pytorch(self)867 @classmethod868 def from_pretrained(self, *args, **kwargs):869 requires_pytorch(self)870class FunnelForSequenceClassification:871 def __init__(self, *args, **kwargs):872 requires_pytorch(self)873 @classmethod874 def from_pretrained(self, *args, **kwargs):875 requires_pytorch(self)876class FunnelForTokenClassification:877 def __init__(self, *args, **kwargs):878 requires_pytorch(self)879 @classmethod880 def from_pretrained(self, *args, **kwargs):881 requires_pytorch(self)882class FunnelModel:883 def __init__(self, *args, **kwargs):884 requires_pytorch(self)885 @classmethod886 def from_pretrained(self, *args, **kwargs):887 requires_pytorch(self)888def load_tf_weights_in_funnel(*args, **kwargs):889 requires_pytorch(load_tf_weights_in_funnel)890GPT2_PRETRAINED_MODEL_ARCHIVE_LIST = None891class GPT2DoubleHeadsModel:892 def __init__(self, *args, **kwargs):893 requires_pytorch(self)894 @classmethod895 def from_pretrained(self, *args, **kwargs):896 requires_pytorch(self)897class GPT2ForSequenceClassification:898 def __init__(self, *args, **kwargs):899 requires_pytorch(self)900 @classmethod901 def from_pretrained(self, *args, **kwargs):902 requires_pytorch(self)903class GPT2LMHeadModel:904 def __init__(self, *args, **kwargs):905 requires_pytorch(self)906 @classmethod907 def from_pretrained(self, *args, **kwargs):908 requires_pytorch(self)909class GPT2Model:910 def __init__(self, *args, **kwargs):911 requires_pytorch(self)912 @classmethod913 def from_pretrained(self, *args, **kwargs):914 requires_pytorch(self)915class GPT2PreTrainedModel:916 def __init__(self, *args, **kwargs):917 requires_pytorch(self)918 @classmethod919 def from_pretrained(self, *args, **kwargs):920 requires_pytorch(self)921def load_tf_weights_in_gpt2(*args, **kwargs):922 requires_pytorch(load_tf_weights_in_gpt2)923GPT_NEO_PRETRAINED_MODEL_ARCHIVE_LIST = None924class GPTNeoForCausalLM:925 def __init__(self, *args, **kwargs):926 requires_pytorch(self)927class GPTNeoModel:928 def __init__(self, *args, **kwargs):929 requires_pytorch(self)930 @classmethod931 def from_pretrained(self, *args, **kwargs):932 requires_pytorch(self)933class GPTNeoPreTrainedModel:934 def __init__(self, *args, **kwargs):935 requires_pytorch(self)936 @classmethod937 def from_pretrained(self, *args, **kwargs):938 requires_pytorch(self)939def load_tf_weights_in_gpt_neo(*args, **kwargs):940 requires_pytorch(load_tf_weights_in_gpt_neo)941IBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None942class IBertForMaskedLM:943 def __init__(self, *args, **kwargs):944 requires_pytorch(self)945 @classmethod946 def from_pretrained(self, *args, **kwargs):947 requires_pytorch(self)948class IBertForMultipleChoice:949 def __init__(self, *args, **kwargs):950 requires_pytorch(self)951 @classmethod952 def from_pretrained(self, *args, **kwargs):953 requires_pytorch(self)954class IBertForQuestionAnswering:955 def __init__(self, *args, **kwargs):956 requires_pytorch(self)957 @classmethod958 def from_pretrained(self, *args, **kwargs):959 requires_pytorch(self)960class IBertForSequenceClassification:961 def __init__(self, *args, **kwargs):962 requires_pytorch(self)963 @classmethod964 def from_pretrained(self, *args, **kwargs):965 requires_pytorch(self)966class IBertForTokenClassification:967 def __init__(self, *args, **kwargs):968 requires_pytorch(self)969 @classmethod970 def from_pretrained(self, *args, **kwargs):971 requires_pytorch(self)972class IBertModel:973 def __init__(self, *args, **kwargs):974 requires_pytorch(self)975 @classmethod976 def from_pretrained(self, *args, **kwargs):977 requires_pytorch(self)978class IBertPreTrainedModel:979 def __init__(self, *args, **kwargs):980 requires_pytorch(self)981 @classmethod982 def from_pretrained(self, *args, **kwargs):983 requires_pytorch(self)984LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST = None985class LayoutLMForMaskedLM:986 def __init__(self, *args, **kwargs):987 requires_pytorch(self)988 @classmethod989 def from_pretrained(self, *args, **kwargs):990 requires_pytorch(self)991class LayoutLMForSequenceClassification:992 def __init__(self, *args, **kwargs):993 requires_pytorch(self)994 @classmethod995 def from_pretrained(self, *args, **kwargs):996 requires_pytorch(self)997class LayoutLMForTokenClassification:998 def __init__(self, *args, **kwargs):999 requires_pytorch(self)1000 @classmethod1001 def from_pretrained(self, *args, **kwargs):1002 requires_pytorch(self)1003class LayoutLMModel:1004 def __init__(self, *args, **kwargs):1005 requires_pytorch(self)1006 @classmethod1007 def from_pretrained(self, *args, **kwargs):1008 requires_pytorch(self)1009LED_PRETRAINED_MODEL_ARCHIVE_LIST = None1010class LEDForConditionalGeneration:1011 def __init__(self, *args, **kwargs):1012 requires_pytorch(self)1013 @classmethod1014 def from_pretrained(self, *args, **kwargs):1015 requires_pytorch(self)1016class LEDForQuestionAnswering:1017 def __init__(self, *args, **kwargs):1018 requires_pytorch(self)1019 @classmethod1020 def from_pretrained(self, *args, **kwargs):1021 requires_pytorch(self)1022class LEDForSequenceClassification:1023 def __init__(self, *args, **kwargs):1024 requires_pytorch(self)1025 @classmethod1026 def from_pretrained(self, *args, **kwargs):1027 requires_pytorch(self)1028class LEDModel:1029 def __init__(self, *args, **kwargs):1030 requires_pytorch(self)1031 @classmethod1032 def from_pretrained(self, *args, **kwargs):1033 requires_pytorch(self)1034LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None1035class LongformerForMaskedLM:1036 def __init__(self, *args, **kwargs):1037 requires_pytorch(self)1038 @classmethod1039 def from_pretrained(self, *args, **kwargs):1040 requires_pytorch(self)1041class LongformerForMultipleChoice:1042 def __init__(self, *args, **kwargs):1043 requires_pytorch(self)1044 @classmethod1045 def from_pretrained(self, *args, **kwargs):1046 requires_pytorch(self)1047class LongformerForQuestionAnswering:1048 def __init__(self, *args, **kwargs):1049 requires_pytorch(self)1050 @classmethod1051 def from_pretrained(self, *args, **kwargs):1052 requires_pytorch(self)1053class LongformerForSequenceClassification:1054 def __init__(self, *args, **kwargs):1055 requires_pytorch(self)1056 @classmethod1057 def from_pretrained(self, *args, **kwargs):1058 requires_pytorch(self)1059class LongformerForTokenClassification:1060 def __init__(self, *args, **kwargs):1061 requires_pytorch(self)1062 @classmethod1063 def from_pretrained(self, *args, **kwargs):1064 requires_pytorch(self)1065class LongformerModel:1066 def __init__(self, *args, **kwargs):1067 requires_pytorch(self)1068 @classmethod1069 def from_pretrained(self, *args, **kwargs):1070 requires_pytorch(self)1071class LongformerSelfAttention:1072 def __init__(self, *args, **kwargs):1073 requires_pytorch(self)1074class LxmertEncoder:1075 def __init__(self, *args, **kwargs):1076 requires_pytorch(self)1077class LxmertForPreTraining:1078 def __init__(self, *args, **kwargs):1079 requires_pytorch(self)1080class LxmertForQuestionAnswering:1081 def __init__(self, *args, **kwargs):1082 requires_pytorch(self)1083 @classmethod1084 def from_pretrained(self, *args, **kwargs):1085 requires_pytorch(self)1086class LxmertModel:1087 def __init__(self, *args, **kwargs):1088 requires_pytorch(self)1089 @classmethod1090 def from_pretrained(self, *args, **kwargs):1091 requires_pytorch(self)1092class LxmertPreTrainedModel:1093 def __init__(self, *args, **kwargs):1094 requires_pytorch(self)1095 @classmethod1096 def from_pretrained(self, *args, **kwargs):1097 requires_pytorch(self)1098class LxmertVisualFeatureEncoder:1099 def __init__(self, *args, **kwargs):1100 requires_pytorch(self)1101class LxmertXLayer:1102 def __init__(self, *args, **kwargs):1103 requires_pytorch(self)1104M2M_100_PRETRAINED_MODEL_ARCHIVE_LIST = None1105class M2M100ForConditionalGeneration:1106 def __init__(self, *args, **kwargs):1107 requires_pytorch(self)1108 @classmethod1109 def from_pretrained(self, *args, **kwargs):1110 requires_pytorch(self)1111class M2M100Model:1112 def __init__(self, *args, **kwargs):1113 requires_pytorch(self)1114 @classmethod1115 def from_pretrained(self, *args, **kwargs):1116 requires_pytorch(self)1117class MarianForCausalLM:1118 def __init__(self, *args, **kwargs):1119 requires_pytorch(self)1120class MarianModel:1121 def __init__(self, *args, **kwargs):1122 requires_pytorch(self)1123 @classmethod1124 def from_pretrained(self, *args, **kwargs):1125 requires_pytorch(self)1126class MarianMTModel:1127 def __init__(self, *args, **kwargs):1128 requires_pytorch(self)1129 @classmethod1130 def from_pretrained(self, *args, **kwargs):1131 requires_pytorch(self)1132class MBartForCausalLM:1133 def __init__(self, *args, **kwargs):1134 requires_pytorch(self)1135class MBartForConditionalGeneration:1136 def __init__(self, *args, **kwargs):1137 requires_pytorch(self)1138 @classmethod1139 def from_pretrained(self, *args, **kwargs):1140 requires_pytorch(self)1141class MBartForQuestionAnswering:1142 def __init__(self, *args, **kwargs):1143 requires_pytorch(self)1144 @classmethod1145 def from_pretrained(self, *args, **kwargs):1146 requires_pytorch(self)1147class MBartForSequenceClassification:1148 def __init__(self, *args, **kwargs):1149 requires_pytorch(self)1150 @classmethod1151 def from_pretrained(self, *args, **kwargs):1152 requires_pytorch(self)1153class MBartModel:1154 def __init__(self, *args, **kwargs):1155 requires_pytorch(self)1156 @classmethod1157 def from_pretrained(self, *args, **kwargs):1158 requires_pytorch(self)1159class MMBTForClassification:1160 def __init__(self, *args, **kwargs):1161 requires_pytorch(self)1162class MMBTModel:1163 def __init__(self, *args, **kwargs):1164 requires_pytorch(self)1165 @classmethod1166 def from_pretrained(self, *args, **kwargs):1167 requires_pytorch(self)1168class ModalEmbeddings:1169 def __init__(self, *args, **kwargs):1170 requires_pytorch(self)1171MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None1172class MobileBertForMaskedLM:1173 def __init__(self, *args, **kwargs):1174 requires_pytorch(self)1175 @classmethod1176 def from_pretrained(self, *args, **kwargs):1177 requires_pytorch(self)1178class MobileBertForMultipleChoice:1179 def __init__(self, *args, **kwargs):1180 requires_pytorch(self)1181 @classmethod1182 def from_pretrained(self, *args, **kwargs):1183 requires_pytorch(self)1184class MobileBertForNextSentencePrediction:1185 def __init__(self, *args, **kwargs):1186 requires_pytorch(self)1187class MobileBertForPreTraining:1188 def __init__(self, *args, **kwargs):1189 requires_pytorch(self)1190class MobileBertForQuestionAnswering:1191 def __init__(self, *args, **kwargs):1192 requires_pytorch(self)1193 @classmethod1194 def from_pretrained(self, *args, **kwargs):1195 requires_pytorch(self)1196class MobileBertForSequenceClassification:1197 def __init__(self, *args, **kwargs):1198 requires_pytorch(self)1199 @classmethod1200 def from_pretrained(self, *args, **kwargs):1201 requires_pytorch(self)1202class MobileBertForTokenClassification:1203 def __init__(self, *args, **kwargs):1204 requires_pytorch(self)1205 @classmethod1206 def from_pretrained(self, *args, **kwargs):1207 requires_pytorch(self)1208class MobileBertLayer:1209 def __init__(self, *args, **kwargs):1210 requires_pytorch(self)1211class MobileBertModel:1212 def __init__(self, *args, **kwargs):1213 requires_pytorch(self)1214 @classmethod1215 def from_pretrained(self, *args, **kwargs):1216 requires_pytorch(self)1217class MobileBertPreTrainedModel:1218 def __init__(self, *args, **kwargs):1219 requires_pytorch(self)1220 @classmethod1221 def from_pretrained(self, *args, **kwargs):1222 requires_pytorch(self)1223def load_tf_weights_in_mobilebert(*args, **kwargs):1224 requires_pytorch(load_tf_weights_in_mobilebert)1225MPNET_PRETRAINED_MODEL_ARCHIVE_LIST = None1226class MPNetForMaskedLM:1227 def __init__(self, *args, **kwargs):1228 requires_pytorch(self)1229 @classmethod1230 def from_pretrained(self, *args, **kwargs):1231 requires_pytorch(self)1232class MPNetForMultipleChoice:1233 def __init__(self, *args, **kwargs):1234 requires_pytorch(self)1235 @classmethod1236 def from_pretrained(self, *args, **kwargs):1237 requires_pytorch(self)1238class MPNetForQuestionAnswering:1239 def __init__(self, *args, **kwargs):1240 requires_pytorch(self)1241 @classmethod1242 def from_pretrained(self, *args, **kwargs):1243 requires_pytorch(self)1244class MPNetForSequenceClassification:1245 def __init__(self, *args, **kwargs):1246 requires_pytorch(self)1247 @classmethod1248 def from_pretrained(self, *args, **kwargs):1249 requires_pytorch(self)1250class MPNetForTokenClassification:1251 def __init__(self, *args, **kwargs):1252 requires_pytorch(self)1253 @classmethod1254 def from_pretrained(self, *args, **kwargs):1255 requires_pytorch(self)1256class MPNetLayer:1257 def __init__(self, *args, **kwargs):1258 requires_pytorch(self)1259class MPNetModel:1260 def __init__(self, *args, **kwargs):1261 requires_pytorch(self)1262 @classmethod1263 def from_pretrained(self, *args, **kwargs):1264 requires_pytorch(self)1265class MPNetPreTrainedModel:1266 def __init__(self, *args, **kwargs):1267 requires_pytorch(self)1268 @classmethod1269 def from_pretrained(self, *args, **kwargs):1270 requires_pytorch(self)1271class MT5EncoderModel:1272 def __init__(self, *args, **kwargs):1273 requires_pytorch(self)1274 @classmethod1275 def from_pretrained(self, *args, **kwargs):1276 requires_pytorch(self)1277class MT5ForConditionalGeneration:1278 def __init__(self, *args, **kwargs):1279 requires_pytorch(self)1280 @classmethod1281 def from_pretrained(self, *args, **kwargs):1282 requires_pytorch(self)1283class MT5Model:1284 def __init__(self, *args, **kwargs):1285 requires_pytorch(self)1286 @classmethod1287 def from_pretrained(self, *args, **kwargs):1288 requires_pytorch(self)1289OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST = None1290class OpenAIGPTDoubleHeadsModel:1291 def __init__(self, *args, **kwargs):1292 requires_pytorch(self)1293 @classmethod1294 def from_pretrained(self, *args, **kwargs):1295 requires_pytorch(self)1296class OpenAIGPTForSequenceClassification:1297 def __init__(self, *args, **kwargs):1298 requires_pytorch(self)1299 @classmethod1300 def from_pretrained(self, *args, **kwargs):1301 requires_pytorch(self)1302class OpenAIGPTLMHeadModel:1303 def __init__(self, *args, **kwargs):1304 requires_pytorch(self)1305 @classmethod1306 def from_pretrained(self, *args, **kwargs):1307 requires_pytorch(self)1308class OpenAIGPTModel:1309 def __init__(self, *args, **kwargs):1310 requires_pytorch(self)1311 @classmethod1312 def from_pretrained(self, *args, **kwargs):1313 requires_pytorch(self)1314class OpenAIGPTPreTrainedModel:1315 def __init__(self, *args, **kwargs):1316 requires_pytorch(self)1317 @classmethod1318 def from_pretrained(self, *args, **kwargs):1319 requires_pytorch(self)1320def load_tf_weights_in_openai_gpt(*args, **kwargs):1321 requires_pytorch(load_tf_weights_in_openai_gpt)1322class PegasusForCausalLM:1323 def __init__(self, *args, **kwargs):1324 requires_pytorch(self)1325class PegasusForConditionalGeneration:1326 def __init__(self, *args, **kwargs):1327 requires_pytorch(self)1328 @classmethod1329 def from_pretrained(self, *args, **kwargs):1330 requires_pytorch(self)1331class PegasusModel:1332 def __init__(self, *args, **kwargs):1333 requires_pytorch(self)1334 @classmethod1335 def from_pretrained(self, *args, **kwargs):1336 requires_pytorch(self)1337PROPHETNET_PRETRAINED_MODEL_ARCHIVE_LIST = None1338class ProphetNetDecoder:1339 def __init__(self, *args, **kwargs):1340 requires_pytorch(self)1341class ProphetNetEncoder:1342 def __init__(self, *args, **kwargs):1343 requires_pytorch(self)1344class ProphetNetForCausalLM:1345 def __init__(self, *args, **kwargs):1346 requires_pytorch(self)1347class ProphetNetForConditionalGeneration:1348 def __init__(self, *args, **kwargs):1349 requires_pytorch(self)1350 @classmethod1351 def from_pretrained(self, *args, **kwargs):1352 requires_pytorch(self)1353class ProphetNetModel:1354 def __init__(self, *args, **kwargs):1355 requires_pytorch(self)1356 @classmethod1357 def from_pretrained(self, *args, **kwargs):1358 requires_pytorch(self)1359class ProphetNetPreTrainedModel:1360 def __init__(self, *args, **kwargs):1361 requires_pytorch(self)1362 @classmethod1363 def from_pretrained(self, *args, **kwargs):1364 requires_pytorch(self)1365class RagModel:1366 def __init__(self, *args, **kwargs):1367 requires_pytorch(self)1368 @classmethod1369 def from_pretrained(self, *args, **kwargs):1370 requires_pytorch(self)1371class RagSequenceForGeneration:1372 def __init__(self, *args, **kwargs):1373 requires_pytorch(self)1374class RagTokenForGeneration:1375 def __init__(self, *args, **kwargs):1376 requires_pytorch(self)1377REFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None1378class ReformerAttention:1379 def __init__(self, *args, **kwargs):1380 requires_pytorch(self)1381class ReformerForMaskedLM:1382 def __init__(self, *args, **kwargs):1383 requires_pytorch(self)1384 @classmethod1385 def from_pretrained(self, *args, **kwargs):1386 requires_pytorch(self)1387class ReformerForQuestionAnswering:1388 def __init__(self, *args, **kwargs):1389 requires_pytorch(self)1390 @classmethod1391 def from_pretrained(self, *args, **kwargs):1392 requires_pytorch(self)1393class ReformerForSequenceClassification:1394 def __init__(self, *args, **kwargs):1395 requires_pytorch(self)1396 @classmethod1397 def from_pretrained(self, *args, **kwargs):1398 requires_pytorch(self)1399class ReformerLayer:1400 def __init__(self, *args, **kwargs):1401 requires_pytorch(self)1402class ReformerModel:1403 def __init__(self, *args, **kwargs):1404 requires_pytorch(self)1405 @classmethod1406 def from_pretrained(self, *args, **kwargs):1407 requires_pytorch(self)1408class ReformerModelWithLMHead:1409 def __init__(self, *args, **kwargs):1410 requires_pytorch(self)1411 @classmethod1412 def from_pretrained(self, *args, **kwargs):1413 requires_pytorch(self)1414RETRIBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None1415class RetriBertModel:1416 def __init__(self, *args, **kwargs):1417 requires_pytorch(self)1418 @classmethod1419 def from_pretrained(self, *args, **kwargs):1420 requires_pytorch(self)1421class RetriBertPreTrainedModel:1422 def __init__(self, *args, **kwargs):1423 requires_pytorch(self)1424 @classmethod1425 def from_pretrained(self, *args, **kwargs):1426 requires_pytorch(self)1427ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None1428class RobertaForCausalLM:1429 def __init__(self, *args, **kwargs):1430 requires_pytorch(self)1431class RobertaForMaskedLM:1432 def __init__(self, *args, **kwargs):1433 requires_pytorch(self)1434 @classmethod1435 def from_pretrained(self, *args, **kwargs):1436 requires_pytorch(self)1437class RobertaForMultipleChoice:1438 def __init__(self, *args, **kwargs):1439 requires_pytorch(self)1440 @classmethod1441 def from_pretrained(self, *args, **kwargs):1442 requires_pytorch(self)1443class RobertaForQuestionAnswering:1444 def __init__(self, *args, **kwargs):1445 requires_pytorch(self)1446 @classmethod1447 def from_pretrained(self, *args, **kwargs):1448 requires_pytorch(self)1449class RobertaForSequenceClassification:1450 def __init__(self, *args, **kwargs):1451 requires_pytorch(self)1452 @classmethod1453 def from_pretrained(self, *args, **kwargs):1454 requires_pytorch(self)1455class RobertaForTokenClassification:1456 def __init__(self, *args, **kwargs):1457 requires_pytorch(self)1458 @classmethod1459 def from_pretrained(self, *args, **kwargs):1460 requires_pytorch(self)1461class RobertaModel:1462 def __init__(self, *args, **kwargs):1463 requires_pytorch(self)1464 @classmethod1465 def from_pretrained(self, *args, **kwargs):1466 requires_pytorch(self)1467SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST = None1468class Speech2TextForConditionalGeneration:1469 def __init__(self, *args, **kwargs):1470 requires_pytorch(self)1471 @classmethod1472 def from_pretrained(self, *args, **kwargs):1473 requires_pytorch(self)1474class Speech2TextModel:1475 def __init__(self, *args, **kwargs):1476 requires_pytorch(self)1477 @classmethod1478 def from_pretrained(self, *args, **kwargs):1479 requires_pytorch(self)1480SQUEEZEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None1481class SqueezeBertForMaskedLM:1482 def __init__(self, *args, **kwargs):1483 requires_pytorch(self)1484 @classmethod1485 def from_pretrained(self, *args, **kwargs):1486 requires_pytorch(self)1487class SqueezeBertForMultipleChoice:1488 def __init__(self, *args, **kwargs):1489 requires_pytorch(self)1490 @classmethod1491 def from_pretrained(self, *args, **kwargs):1492 requires_pytorch(self)1493class SqueezeBertForQuestionAnswering:1494 def __init__(self, *args, **kwargs):1495 requires_pytorch(self)1496 @classmethod1497 def from_pretrained(self, *args, **kwargs):1498 requires_pytorch(self)1499class SqueezeBertForSequenceClassification:1500 def __init__(self, *args, **kwargs):1501 requires_pytorch(self)1502 @classmethod1503 def from_pretrained(self, *args, **kwargs):1504 requires_pytorch(self)1505class SqueezeBertForTokenClassification:1506 def __init__(self, *args, **kwargs):1507 requires_pytorch(self)1508 @classmethod1509 def from_pretrained(self, *args, **kwargs):1510 requires_pytorch(self)1511class SqueezeBertModel:1512 def __init__(self, *args, **kwargs):1513 requires_pytorch(self)1514 @classmethod1515 def from_pretrained(self, *args, **kwargs):1516 requires_pytorch(self)1517class SqueezeBertModule:1518 def __init__(self, *args, **kwargs):1519 requires_pytorch(self)1520class SqueezeBertPreTrainedModel:1521 def __init__(self, *args, **kwargs):1522 requires_pytorch(self)1523 @classmethod1524 def from_pretrained(self, *args, **kwargs):1525 requires_pytorch(self)1526T5_PRETRAINED_MODEL_ARCHIVE_LIST = None1527class T5EncoderModel:1528 def __init__(self, *args, **kwargs):1529 requires_pytorch(self)1530 @classmethod1531 def from_pretrained(self, *args, **kwargs):1532 requires_pytorch(self)1533class T5ForConditionalGeneration:1534 def __init__(self, *args, **kwargs):1535 requires_pytorch(self)1536 @classmethod1537 def from_pretrained(self, *args, **kwargs):1538 requires_pytorch(self)1539class T5Model:1540 def __init__(self, *args, **kwargs):1541 requires_pytorch(self)1542 @classmethod1543 def from_pretrained(self, *args, **kwargs):1544 requires_pytorch(self)1545class T5PreTrainedModel:1546 def __init__(self, *args, **kwargs):1547 requires_pytorch(self)1548 @classmethod1549 def from_pretrained(self, *args, **kwargs):1550 requires_pytorch(self)1551def load_tf_weights_in_t5(*args, **kwargs):1552 requires_pytorch(load_tf_weights_in_t5)1553TAPAS_PRETRAINED_MODEL_ARCHIVE_LIST = None1554class TapasForMaskedLM:1555 def __init__(self, *args, **kwargs):1556 requires_pytorch(self)1557 @classmethod1558 def from_pretrained(self, *args, **kwargs):1559 requires_pytorch(self)1560class TapasForQuestionAnswering:1561 def __init__(self, *args, **kwargs):1562 requires_pytorch(self)1563 @classmethod1564 def from_pretrained(self, *args, **kwargs):1565 requires_pytorch(self)1566class TapasForSequenceClassification:1567 def __init__(self, *args, **kwargs):1568 requires_pytorch(self)1569 @classmethod1570 def from_pretrained(self, *args, **kwargs):1571 requires_pytorch(self)1572class TapasModel:1573 def __init__(self, *args, **kwargs):1574 requires_pytorch(self)1575 @classmethod1576 def from_pretrained(self, *args, **kwargs):1577 requires_pytorch(self)1578TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LIST = None1579class AdaptiveEmbedding:1580 def __init__(self, *args, **kwargs):1581 requires_pytorch(self)1582class TransfoXLForSequenceClassification:1583 def __init__(self, *args, **kwargs):1584 requires_pytorch(self)1585 @classmethod1586 def from_pretrained(self, *args, **kwargs):1587 requires_pytorch(self)1588class TransfoXLLMHeadModel:1589 def __init__(self, *args, **kwargs):1590 requires_pytorch(self)1591 @classmethod1592 def from_pretrained(self, *args, **kwargs):1593 requires_pytorch(self)1594class TransfoXLModel:1595 def __init__(self, *args, **kwargs):1596 requires_pytorch(self)1597 @classmethod1598 def from_pretrained(self, *args, **kwargs):1599 requires_pytorch(self)1600class TransfoXLPreTrainedModel:1601 def __init__(self, *args, **kwargs):1602 requires_pytorch(self)1603 @classmethod1604 def from_pretrained(self, *args, **kwargs):1605 requires_pytorch(self)1606def load_tf_weights_in_transfo_xl(*args, **kwargs):1607 requires_pytorch(load_tf_weights_in_transfo_xl)1608VIT_PRETRAINED_MODEL_ARCHIVE_LIST = None1609class ViTForImageClassification:1610 def __init__(self, *args, **kwargs):1611 requires_pytorch(self)1612class ViTModel:1613 def __init__(self, *args, **kwargs):1614 requires_pytorch(self)1615 @classmethod1616 def from_pretrained(self, *args, **kwargs):1617 requires_pytorch(self)1618class ViTPreTrainedModel:1619 def __init__(self, *args, **kwargs):1620 requires_pytorch(self)1621 @classmethod1622 def from_pretrained(self, *args, **kwargs):1623 requires_pytorch(self)1624WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST = None1625class Wav2Vec2ForCTC:1626 def __init__(self, *args, **kwargs):1627 requires_pytorch(self)1628class Wav2Vec2ForMaskedLM:1629 def __init__(self, *args, **kwargs):1630 requires_pytorch(self)1631 @classmethod1632 def from_pretrained(self, *args, **kwargs):1633 requires_pytorch(self)1634class Wav2Vec2Model:1635 def __init__(self, *args, **kwargs):1636 requires_pytorch(self)1637 @classmethod1638 def from_pretrained(self, *args, **kwargs):1639 requires_pytorch(self)1640class Wav2Vec2PreTrainedModel:1641 def __init__(self, *args, **kwargs):1642 requires_pytorch(self)1643 @classmethod1644 def from_pretrained(self, *args, **kwargs):1645 requires_pytorch(self)1646XLM_PRETRAINED_MODEL_ARCHIVE_LIST = None1647class XLMForMultipleChoice:1648 def __init__(self, *args, **kwargs):1649 requires_pytorch(self)1650 @classmethod1651 def from_pretrained(self, *args, **kwargs):1652 requires_pytorch(self)1653class XLMForQuestionAnswering:1654 def __init__(self, *args, **kwargs):1655 requires_pytorch(self)1656 @classmethod1657 def from_pretrained(self, *args, **kwargs):1658 requires_pytorch(self)1659class XLMForQuestionAnsweringSimple:1660 def __init__(self, *args, **kwargs):1661 requires_pytorch(self)1662 @classmethod1663 def from_pretrained(self, *args, **kwargs):1664 requires_pytorch(self)1665class XLMForSequenceClassification:1666 def __init__(self, *args, **kwargs):1667 requires_pytorch(self)1668 @classmethod1669 def from_pretrained(self, *args, **kwargs):1670 requires_pytorch(self)1671class XLMForTokenClassification:1672 def __init__(self, *args, **kwargs):1673 requires_pytorch(self)1674 @classmethod1675 def from_pretrained(self, *args, **kwargs):1676 requires_pytorch(self)1677class XLMModel:1678 def __init__(self, *args, **kwargs):1679 requires_pytorch(self)1680 @classmethod1681 def from_pretrained(self, *args, **kwargs):1682 requires_pytorch(self)1683class XLMPreTrainedModel:1684 def __init__(self, *args, **kwargs):1685 requires_pytorch(self)1686 @classmethod1687 def from_pretrained(self, *args, **kwargs):1688 requires_pytorch(self)1689class XLMWithLMHeadModel:1690 def __init__(self, *args, **kwargs):1691 requires_pytorch(self)1692 @classmethod1693 def from_pretrained(self, *args, **kwargs):1694 requires_pytorch(self)1695XLM_PROPHETNET_PRETRAINED_MODEL_ARCHIVE_LIST = None1696class XLMProphetNetDecoder:1697 def __init__(self, *args, **kwargs):1698 requires_pytorch(self)1699class XLMProphetNetEncoder:1700 def __init__(self, *args, **kwargs):1701 requires_pytorch(self)1702class XLMProphetNetForCausalLM:1703 def __init__(self, *args, **kwargs):1704 requires_pytorch(self)1705class XLMProphetNetForConditionalGeneration:1706 def __init__(self, *args, **kwargs):1707 requires_pytorch(self)1708 @classmethod1709 def from_pretrained(self, *args, **kwargs):1710 requires_pytorch(self)1711class XLMProphetNetModel:1712 def __init__(self, *args, **kwargs):1713 requires_pytorch(self)1714 @classmethod1715 def from_pretrained(self, *args, **kwargs):1716 requires_pytorch(self)1717XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None1718class XLMRobertaForCausalLM:1719 def __init__(self, *args, **kwargs):1720 requires_pytorch(self)1721class XLMRobertaForMaskedLM:1722 def __init__(self, *args, **kwargs):1723 requires_pytorch(self)1724 @classmethod1725 def from_pretrained(self, *args, **kwargs):1726 requires_pytorch(self)1727class XLMRobertaForMultipleChoice:1728 def __init__(self, *args, **kwargs):1729 requires_pytorch(self)1730 @classmethod1731 def from_pretrained(self, *args, **kwargs):1732 requires_pytorch(self)1733class XLMRobertaForQuestionAnswering:1734 def __init__(self, *args, **kwargs):1735 requires_pytorch(self)1736 @classmethod1737 def from_pretrained(self, *args, **kwargs):1738 requires_pytorch(self)1739class XLMRobertaForSequenceClassification:1740 def __init__(self, *args, **kwargs):1741 requires_pytorch(self)1742 @classmethod1743 def from_pretrained(self, *args, **kwargs):1744 requires_pytorch(self)1745class XLMRobertaForTokenClassification:1746 def __init__(self, *args, **kwargs):1747 requires_pytorch(self)1748 @classmethod1749 def from_pretrained(self, *args, **kwargs):1750 requires_pytorch(self)1751class XLMRobertaModel:1752 def __init__(self, *args, **kwargs):1753 requires_pytorch(self)1754 @classmethod1755 def from_pretrained(self, *args, **kwargs):1756 requires_pytorch(self)1757XLNET_PRETRAINED_MODEL_ARCHIVE_LIST = None1758class XLNetForMultipleChoice:1759 def __init__(self, *args, **kwargs):1760 requires_pytorch(self)1761 @classmethod1762 def from_pretrained(self, *args, **kwargs):1763 requires_pytorch(self)1764class XLNetForQuestionAnswering:1765 def __init__(self, *args, **kwargs):1766 requires_pytorch(self)1767 @classmethod1768 def from_pretrained(self, *args, **kwargs):1769 requires_pytorch(self)1770class XLNetForQuestionAnsweringSimple:1771 def __init__(self, *args, **kwargs):1772 requires_pytorch(self)1773 @classmethod1774 def from_pretrained(self, *args, **kwargs):1775 requires_pytorch(self)1776class XLNetForSequenceClassification:1777 def __init__(self, *args, **kwargs):1778 requires_pytorch(self)1779 @classmethod1780 def from_pretrained(self, *args, **kwargs):1781 requires_pytorch(self)1782class XLNetForTokenClassification:1783 def __init__(self, *args, **kwargs):1784 requires_pytorch(self)1785 @classmethod1786 def from_pretrained(self, *args, **kwargs):1787 requires_pytorch(self)1788class XLNetLMHeadModel:1789 def __init__(self, *args, **kwargs):1790 requires_pytorch(self)1791 @classmethod1792 def from_pretrained(self, *args, **kwargs):1793 requires_pytorch(self)1794class XLNetModel:1795 def __init__(self, *args, **kwargs):1796 requires_pytorch(self)1797 @classmethod1798 def from_pretrained(self, *args, **kwargs):1799 requires_pytorch(self)1800class XLNetPreTrainedModel:1801 def __init__(self, *args, **kwargs):1802 requires_pytorch(self)1803 @classmethod1804 def from_pretrained(self, *args, **kwargs):1805 requires_pytorch(self)1806def load_tf_weights_in_xlnet(*args, **kwargs):1807 requires_pytorch(load_tf_weights_in_xlnet)1808class Adafactor:1809 def __init__(self, *args, **kwargs):1810 requires_pytorch(self)1811class AdamW:1812 def __init__(self, *args, **kwargs):1813 requires_pytorch(self)1814def get_constant_schedule(*args, **kwargs):1815 requires_pytorch(get_constant_schedule)1816def get_constant_schedule_with_warmup(*args, **kwargs):1817 requires_pytorch(get_constant_schedule_with_warmup)1818def get_cosine_schedule_with_warmup(*args, **kwargs):1819 requires_pytorch(get_cosine_schedule_with_warmup)1820def get_cosine_with_hard_restarts_schedule_with_warmup(*args, **kwargs):1821 requires_pytorch(get_cosine_with_hard_restarts_schedule_with_warmup)1822def get_linear_schedule_with_warmup(*args, **kwargs):1823 requires_pytorch(get_linear_schedule_with_warmup)1824def get_polynomial_decay_schedule_with_warmup(*args, **kwargs):1825 requires_pytorch(get_polynomial_decay_schedule_with_warmup)1826def get_scheduler(*args, **kwargs):1827 requires_pytorch(get_scheduler)1828class Trainer:1829 def __init__(self, *args, **kwargs):1830 requires_pytorch(self)1831def torch_distributed_zero_first(*args, **kwargs):1832 requires_pytorch(torch_distributed_zero_first)1833class Seq2SeqTrainer:1834 def __init__(self, *args, **kwargs):...

Full Screen

Full Screen

dummy_tf_objects.py

Source:dummy_tf_objects.py Github

copy

Full Screen

1# This file is autogenerated by the command `make fix-copies`, do not edit.2from ..file_utils import requires_tf3class TensorFlowBenchmarkArguments:4 def __init__(self, *args, **kwargs):5 requires_tf(self)6class TensorFlowBenchmark:7 def __init__(self, *args, **kwargs):8 requires_tf(self)9def tf_top_k_top_p_filtering(*args, **kwargs):10 requires_tf(tf_top_k_top_p_filtering)11TF_LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST = None12class TFLayoutLMForMaskedLM:13 def __init__(self, *args, **kwargs):14 requires_tf(self)15 @classmethod16 def from_pretrained(self, *args, **kwargs):17 requires_tf(self)18class TFLayoutLMForSequenceClassification:19 def __init__(self, *args, **kwargs):20 requires_tf(self)21 @classmethod22 def from_pretrained(self, *args, **kwargs):23 requires_tf(self)24class TFLayoutLMForTokenClassification:25 def __init__(self, *args, **kwargs):26 requires_tf(self)27 @classmethod28 def from_pretrained(self, *args, **kwargs):29 requires_tf(self)30class TFLayoutLMMainLayer:31 def __init__(self, *args, **kwargs):32 requires_tf(self)33class TFLayoutLMModel:34 def __init__(self, *args, **kwargs):35 requires_tf(self)36 @classmethod37 def from_pretrained(self, *args, **kwargs):38 requires_tf(self)39class TFLayoutLMPreTrainedModel:40 def __init__(self, *args, **kwargs):41 requires_tf(self)42 @classmethod43 def from_pretrained(self, *args, **kwargs):44 requires_tf(self)45class TFPreTrainedModel:46 def __init__(self, *args, **kwargs):47 requires_tf(self)48 @classmethod49 def from_pretrained(self, *args, **kwargs):50 requires_tf(self)51class TFSequenceSummary:52 def __init__(self, *args, **kwargs):53 requires_tf(self)54class TFSharedEmbeddings:55 def __init__(self, *args, **kwargs):56 requires_tf(self)57def shape_list(*args, **kwargs):58 requires_tf(shape_list)59TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None60class TFAlbertForMaskedLM:61 def __init__(self, *args, **kwargs):62 requires_tf(self)63 @classmethod64 def from_pretrained(self, *args, **kwargs):65 requires_tf(self)66class TFAlbertForMultipleChoice:67 def __init__(self, *args, **kwargs):68 requires_tf(self)69 @classmethod70 def from_pretrained(self, *args, **kwargs):71 requires_tf(self)72class TFAlbertForPreTraining:73 def __init__(self, *args, **kwargs):74 requires_tf(self)75class TFAlbertForQuestionAnswering:76 def __init__(self, *args, **kwargs):77 requires_tf(self)78 @classmethod79 def from_pretrained(self, *args, **kwargs):80 requires_tf(self)81class TFAlbertForSequenceClassification:82 def __init__(self, *args, **kwargs):83 requires_tf(self)84 @classmethod85 def from_pretrained(self, *args, **kwargs):86 requires_tf(self)87class TFAlbertForTokenClassification:88 def __init__(self, *args, **kwargs):89 requires_tf(self)90 @classmethod91 def from_pretrained(self, *args, **kwargs):92 requires_tf(self)93class TFAlbertMainLayer:94 def __init__(self, *args, **kwargs):95 requires_tf(self)96class TFAlbertModel:97 def __init__(self, *args, **kwargs):98 requires_tf(self)99 @classmethod100 def from_pretrained(self, *args, **kwargs):101 requires_tf(self)102class TFAlbertPreTrainedModel:103 def __init__(self, *args, **kwargs):104 requires_tf(self)105 @classmethod106 def from_pretrained(self, *args, **kwargs):107 requires_tf(self)108TF_MODEL_FOR_CAUSAL_LM_MAPPING = None109TF_MODEL_FOR_MASKED_LM_MAPPING = None110TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING = None111TF_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING = None112TF_MODEL_FOR_PRETRAINING_MAPPING = None113TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING = None114TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING = None115TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = None116TF_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING = None117TF_MODEL_MAPPING = None118TF_MODEL_WITH_LM_HEAD_MAPPING = None119class TFAutoModel:120 def __init__(self, *args, **kwargs):121 requires_tf(self)122 @classmethod123 def from_pretrained(self, *args, **kwargs):124 requires_tf(self)125class TFAutoModelForCausalLM:126 def __init__(self, *args, **kwargs):127 requires_tf(self)128 @classmethod129 def from_pretrained(self, *args, **kwargs):130 requires_tf(self)131class TFAutoModelForMaskedLM:132 def __init__(self, *args, **kwargs):133 requires_tf(self)134 @classmethod135 def from_pretrained(self, *args, **kwargs):136 requires_tf(self)137class TFAutoModelForMultipleChoice:138 def __init__(self, *args, **kwargs):139 requires_tf(self)140 @classmethod141 def from_pretrained(self, *args, **kwargs):142 requires_tf(self)143class TFAutoModelForPreTraining:144 def __init__(self, *args, **kwargs):145 requires_tf(self)146 @classmethod147 def from_pretrained(self, *args, **kwargs):148 requires_tf(self)149class TFAutoModelForQuestionAnswering:150 def __init__(self, *args, **kwargs):151 requires_tf(self)152 @classmethod153 def from_pretrained(self, *args, **kwargs):154 requires_tf(self)155class TFAutoModelForSeq2SeqLM:156 def __init__(self, *args, **kwargs):157 requires_tf(self)158 @classmethod159 def from_pretrained(self, *args, **kwargs):160 requires_tf(self)161class TFAutoModelForSequenceClassification:162 def __init__(self, *args, **kwargs):163 requires_tf(self)164 @classmethod165 def from_pretrained(self, *args, **kwargs):166 requires_tf(self)167class TFAutoModelForTokenClassification:168 def __init__(self, *args, **kwargs):169 requires_tf(self)170 @classmethod171 def from_pretrained(self, *args, **kwargs):172 requires_tf(self)173class TFAutoModelWithLMHead:174 def __init__(self, *args, **kwargs):175 requires_tf(self)176 @classmethod177 def from_pretrained(self, *args, **kwargs):178 requires_tf(self)179class TFBartForConditionalGeneration:180 def __init__(self, *args, **kwargs):181 requires_tf(self)182 @classmethod183 def from_pretrained(self, *args, **kwargs):184 requires_tf(self)185class TFBartModel:186 def __init__(self, *args, **kwargs):187 requires_tf(self)188 @classmethod189 def from_pretrained(self, *args, **kwargs):190 requires_tf(self)191class TFBartPretrainedModel:192 def __init__(self, *args, **kwargs):193 requires_tf(self)194 @classmethod195 def from_pretrained(self, *args, **kwargs):196 requires_tf(self)197TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None198class TFBertEmbeddings:199 def __init__(self, *args, **kwargs):200 requires_tf(self)201class TFBertForMaskedLM:202 def __init__(self, *args, **kwargs):203 requires_tf(self)204 @classmethod205 def from_pretrained(self, *args, **kwargs):206 requires_tf(self)207class TFBertForMultipleChoice:208 def __init__(self, *args, **kwargs):209 requires_tf(self)210 @classmethod211 def from_pretrained(self, *args, **kwargs):212 requires_tf(self)213class TFBertForNextSentencePrediction:214 def __init__(self, *args, **kwargs):215 requires_tf(self)216class TFBertForPreTraining:217 def __init__(self, *args, **kwargs):218 requires_tf(self)219class TFBertForQuestionAnswering:220 def __init__(self, *args, **kwargs):221 requires_tf(self)222 @classmethod223 def from_pretrained(self, *args, **kwargs):224 requires_tf(self)225class TFBertForSequenceClassification:226 def __init__(self, *args, **kwargs):227 requires_tf(self)228 @classmethod229 def from_pretrained(self, *args, **kwargs):230 requires_tf(self)231class TFBertForTokenClassification:232 def __init__(self, *args, **kwargs):233 requires_tf(self)234 @classmethod235 def from_pretrained(self, *args, **kwargs):236 requires_tf(self)237class TFBertLMHeadModel:238 def __init__(self, *args, **kwargs):239 requires_tf(self)240 @classmethod241 def from_pretrained(self, *args, **kwargs):242 requires_tf(self)243class TFBertMainLayer:244 def __init__(self, *args, **kwargs):245 requires_tf(self)246class TFBertModel:247 def __init__(self, *args, **kwargs):248 requires_tf(self)249 @classmethod250 def from_pretrained(self, *args, **kwargs):251 requires_tf(self)252class TFBertPreTrainedModel:253 def __init__(self, *args, **kwargs):254 requires_tf(self)255 @classmethod256 def from_pretrained(self, *args, **kwargs):257 requires_tf(self)258class TFBlenderbotForConditionalGeneration:259 def __init__(self, *args, **kwargs):260 requires_tf(self)261 @classmethod262 def from_pretrained(self, *args, **kwargs):263 requires_tf(self)264class TFBlenderbotModel:265 def __init__(self, *args, **kwargs):266 requires_tf(self)267 @classmethod268 def from_pretrained(self, *args, **kwargs):269 requires_tf(self)270class TFBlenderbotSmallForConditionalGeneration:271 def __init__(self, *args, **kwargs):272 requires_tf(self)273 @classmethod274 def from_pretrained(self, *args, **kwargs):275 requires_tf(self)276class TFBlenderbotSmallModel:277 def __init__(self, *args, **kwargs):278 requires_tf(self)279 @classmethod280 def from_pretrained(self, *args, **kwargs):281 requires_tf(self)282TF_CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None283class TFCamembertForMaskedLM:284 def __init__(self, *args, **kwargs):285 requires_tf(self)286 @classmethod287 def from_pretrained(self, *args, **kwargs):288 requires_tf(self)289class TFCamembertForMultipleChoice:290 def __init__(self, *args, **kwargs):291 requires_tf(self)292 @classmethod293 def from_pretrained(self, *args, **kwargs):294 requires_tf(self)295class TFCamembertForQuestionAnswering:296 def __init__(self, *args, **kwargs):297 requires_tf(self)298 @classmethod299 def from_pretrained(self, *args, **kwargs):300 requires_tf(self)301class TFCamembertForSequenceClassification:302 def __init__(self, *args, **kwargs):303 requires_tf(self)304 @classmethod305 def from_pretrained(self, *args, **kwargs):306 requires_tf(self)307class TFCamembertForTokenClassification:308 def __init__(self, *args, **kwargs):309 requires_tf(self)310 @classmethod311 def from_pretrained(self, *args, **kwargs):312 requires_tf(self)313class TFCamembertModel:314 def __init__(self, *args, **kwargs):315 requires_tf(self)316 @classmethod317 def from_pretrained(self, *args, **kwargs):318 requires_tf(self)319TF_CONVBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None320class TFConvBertForMaskedLM:321 def __init__(self, *args, **kwargs):322 requires_tf(self)323 @classmethod324 def from_pretrained(self, *args, **kwargs):325 requires_tf(self)326class TFConvBertForMultipleChoice:327 def __init__(self, *args, **kwargs):328 requires_tf(self)329 @classmethod330 def from_pretrained(self, *args, **kwargs):331 requires_tf(self)332class TFConvBertForQuestionAnswering:333 def __init__(self, *args, **kwargs):334 requires_tf(self)335 @classmethod336 def from_pretrained(self, *args, **kwargs):337 requires_tf(self)338class TFConvBertForSequenceClassification:339 def __init__(self, *args, **kwargs):340 requires_tf(self)341 @classmethod342 def from_pretrained(self, *args, **kwargs):343 requires_tf(self)344class TFConvBertForTokenClassification:345 def __init__(self, *args, **kwargs):346 requires_tf(self)347 @classmethod348 def from_pretrained(self, *args, **kwargs):349 requires_tf(self)350class TFConvBertLayer:351 def __init__(self, *args, **kwargs):352 requires_tf(self)353class TFConvBertModel:354 def __init__(self, *args, **kwargs):355 requires_tf(self)356 @classmethod357 def from_pretrained(self, *args, **kwargs):358 requires_tf(self)359class TFConvBertPreTrainedModel:360 def __init__(self, *args, **kwargs):361 requires_tf(self)362 @classmethod363 def from_pretrained(self, *args, **kwargs):364 requires_tf(self)365TF_CTRL_PRETRAINED_MODEL_ARCHIVE_LIST = None366class TFCTRLForSequenceClassification:367 def __init__(self, *args, **kwargs):368 requires_tf(self)369 @classmethod370 def from_pretrained(self, *args, **kwargs):371 requires_tf(self)372class TFCTRLLMHeadModel:373 def __init__(self, *args, **kwargs):374 requires_tf(self)375 @classmethod376 def from_pretrained(self, *args, **kwargs):377 requires_tf(self)378class TFCTRLModel:379 def __init__(self, *args, **kwargs):380 requires_tf(self)381 @classmethod382 def from_pretrained(self, *args, **kwargs):383 requires_tf(self)384class TFCTRLPreTrainedModel:385 def __init__(self, *args, **kwargs):386 requires_tf(self)387 @classmethod388 def from_pretrained(self, *args, **kwargs):389 requires_tf(self)390TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None391class TFDistilBertForMaskedLM:392 def __init__(self, *args, **kwargs):393 requires_tf(self)394 @classmethod395 def from_pretrained(self, *args, **kwargs):396 requires_tf(self)397class TFDistilBertForMultipleChoice:398 def __init__(self, *args, **kwargs):399 requires_tf(self)400 @classmethod401 def from_pretrained(self, *args, **kwargs):402 requires_tf(self)403class TFDistilBertForQuestionAnswering:404 def __init__(self, *args, **kwargs):405 requires_tf(self)406 @classmethod407 def from_pretrained(self, *args, **kwargs):408 requires_tf(self)409class TFDistilBertForSequenceClassification:410 def __init__(self, *args, **kwargs):411 requires_tf(self)412 @classmethod413 def from_pretrained(self, *args, **kwargs):414 requires_tf(self)415class TFDistilBertForTokenClassification:416 def __init__(self, *args, **kwargs):417 requires_tf(self)418 @classmethod419 def from_pretrained(self, *args, **kwargs):420 requires_tf(self)421class TFDistilBertMainLayer:422 def __init__(self, *args, **kwargs):423 requires_tf(self)424class TFDistilBertModel:425 def __init__(self, *args, **kwargs):426 requires_tf(self)427 @classmethod428 def from_pretrained(self, *args, **kwargs):429 requires_tf(self)430class TFDistilBertPreTrainedModel:431 def __init__(self, *args, **kwargs):432 requires_tf(self)433 @classmethod434 def from_pretrained(self, *args, **kwargs):435 requires_tf(self)436TF_DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None437TF_DPR_QUESTION_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None438TF_DPR_READER_PRETRAINED_MODEL_ARCHIVE_LIST = None439class TFDPRContextEncoder:440 def __init__(self, *args, **kwargs):441 requires_tf(self)442class TFDPRPretrainedContextEncoder:443 def __init__(self, *args, **kwargs):444 requires_tf(self)445class TFDPRPretrainedQuestionEncoder:446 def __init__(self, *args, **kwargs):447 requires_tf(self)448class TFDPRPretrainedReader:449 def __init__(self, *args, **kwargs):450 requires_tf(self)451class TFDPRQuestionEncoder:452 def __init__(self, *args, **kwargs):453 requires_tf(self)454class TFDPRReader:455 def __init__(self, *args, **kwargs):456 requires_tf(self)457TF_ELECTRA_PRETRAINED_MODEL_ARCHIVE_LIST = None458class TFElectraForMaskedLM:459 def __init__(self, *args, **kwargs):460 requires_tf(self)461 @classmethod462 def from_pretrained(self, *args, **kwargs):463 requires_tf(self)464class TFElectraForMultipleChoice:465 def __init__(self, *args, **kwargs):466 requires_tf(self)467 @classmethod468 def from_pretrained(self, *args, **kwargs):469 requires_tf(self)470class TFElectraForPreTraining:471 def __init__(self, *args, **kwargs):472 requires_tf(self)473class TFElectraForQuestionAnswering:474 def __init__(self, *args, **kwargs):475 requires_tf(self)476 @classmethod477 def from_pretrained(self, *args, **kwargs):478 requires_tf(self)479class TFElectraForSequenceClassification:480 def __init__(self, *args, **kwargs):481 requires_tf(self)482 @classmethod483 def from_pretrained(self, *args, **kwargs):484 requires_tf(self)485class TFElectraForTokenClassification:486 def __init__(self, *args, **kwargs):487 requires_tf(self)488 @classmethod489 def from_pretrained(self, *args, **kwargs):490 requires_tf(self)491class TFElectraModel:492 def __init__(self, *args, **kwargs):493 requires_tf(self)494 @classmethod495 def from_pretrained(self, *args, **kwargs):496 requires_tf(self)497class TFElectraPreTrainedModel:498 def __init__(self, *args, **kwargs):499 requires_tf(self)500 @classmethod501 def from_pretrained(self, *args, **kwargs):502 requires_tf(self)503TF_FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None504class TFFlaubertForMultipleChoice:505 def __init__(self, *args, **kwargs):506 requires_tf(self)507 @classmethod508 def from_pretrained(self, *args, **kwargs):509 requires_tf(self)510class TFFlaubertForQuestionAnsweringSimple:511 def __init__(self, *args, **kwargs):512 requires_tf(self)513 @classmethod514 def from_pretrained(self, *args, **kwargs):515 requires_tf(self)516class TFFlaubertForSequenceClassification:517 def __init__(self, *args, **kwargs):518 requires_tf(self)519 @classmethod520 def from_pretrained(self, *args, **kwargs):521 requires_tf(self)522class TFFlaubertForTokenClassification:523 def __init__(self, *args, **kwargs):524 requires_tf(self)525 @classmethod526 def from_pretrained(self, *args, **kwargs):527 requires_tf(self)528class TFFlaubertModel:529 def __init__(self, *args, **kwargs):530 requires_tf(self)531 @classmethod532 def from_pretrained(self, *args, **kwargs):533 requires_tf(self)534class TFFlaubertWithLMHeadModel:535 def __init__(self, *args, **kwargs):536 requires_tf(self)537 @classmethod538 def from_pretrained(self, *args, **kwargs):539 requires_tf(self)540TF_FUNNEL_PRETRAINED_MODEL_ARCHIVE_LIST = None541class TFFunnelBaseModel:542 def __init__(self, *args, **kwargs):543 requires_tf(self)544 @classmethod545 def from_pretrained(self, *args, **kwargs):546 requires_tf(self)547class TFFunnelForMaskedLM:548 def __init__(self, *args, **kwargs):549 requires_tf(self)550 @classmethod551 def from_pretrained(self, *args, **kwargs):552 requires_tf(self)553class TFFunnelForMultipleChoice:554 def __init__(self, *args, **kwargs):555 requires_tf(self)556 @classmethod557 def from_pretrained(self, *args, **kwargs):558 requires_tf(self)559class TFFunnelForPreTraining:560 def __init__(self, *args, **kwargs):561 requires_tf(self)562class TFFunnelForQuestionAnswering:563 def __init__(self, *args, **kwargs):564 requires_tf(self)565 @classmethod566 def from_pretrained(self, *args, **kwargs):567 requires_tf(self)568class TFFunnelForSequenceClassification:569 def __init__(self, *args, **kwargs):570 requires_tf(self)571 @classmethod572 def from_pretrained(self, *args, **kwargs):573 requires_tf(self)574class TFFunnelForTokenClassification:575 def __init__(self, *args, **kwargs):576 requires_tf(self)577 @classmethod578 def from_pretrained(self, *args, **kwargs):579 requires_tf(self)580class TFFunnelModel:581 def __init__(self, *args, **kwargs):582 requires_tf(self)583 @classmethod584 def from_pretrained(self, *args, **kwargs):585 requires_tf(self)586TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST = None587class TFGPT2DoubleHeadsModel:588 def __init__(self, *args, **kwargs):589 requires_tf(self)590 @classmethod591 def from_pretrained(self, *args, **kwargs):592 requires_tf(self)593class TFGPT2ForSequenceClassification:594 def __init__(self, *args, **kwargs):595 requires_tf(self)596 @classmethod597 def from_pretrained(self, *args, **kwargs):598 requires_tf(self)599class TFGPT2LMHeadModel:600 def __init__(self, *args, **kwargs):601 requires_tf(self)602 @classmethod603 def from_pretrained(self, *args, **kwargs):604 requires_tf(self)605class TFGPT2MainLayer:606 def __init__(self, *args, **kwargs):607 requires_tf(self)608class TFGPT2Model:609 def __init__(self, *args, **kwargs):610 requires_tf(self)611 @classmethod612 def from_pretrained(self, *args, **kwargs):613 requires_tf(self)614class TFGPT2PreTrainedModel:615 def __init__(self, *args, **kwargs):616 requires_tf(self)617 @classmethod618 def from_pretrained(self, *args, **kwargs):619 requires_tf(self)620class TFLEDForConditionalGeneration:621 def __init__(self, *args, **kwargs):622 requires_tf(self)623 @classmethod624 def from_pretrained(self, *args, **kwargs):625 requires_tf(self)626class TFLEDModel:627 def __init__(self, *args, **kwargs):628 requires_tf(self)629 @classmethod630 def from_pretrained(self, *args, **kwargs):631 requires_tf(self)632class TFLEDPreTrainedModel:633 def __init__(self, *args, **kwargs):634 requires_tf(self)635 @classmethod636 def from_pretrained(self, *args, **kwargs):637 requires_tf(self)638TF_LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None639class TFLongformerForMaskedLM:640 def __init__(self, *args, **kwargs):641 requires_tf(self)642 @classmethod643 def from_pretrained(self, *args, **kwargs):644 requires_tf(self)645class TFLongformerForMultipleChoice:646 def __init__(self, *args, **kwargs):647 requires_tf(self)648 @classmethod649 def from_pretrained(self, *args, **kwargs):650 requires_tf(self)651class TFLongformerForQuestionAnswering:652 def __init__(self, *args, **kwargs):653 requires_tf(self)654 @classmethod655 def from_pretrained(self, *args, **kwargs):656 requires_tf(self)657class TFLongformerForSequenceClassification:658 def __init__(self, *args, **kwargs):659 requires_tf(self)660 @classmethod661 def from_pretrained(self, *args, **kwargs):662 requires_tf(self)663class TFLongformerForTokenClassification:664 def __init__(self, *args, **kwargs):665 requires_tf(self)666 @classmethod667 def from_pretrained(self, *args, **kwargs):668 requires_tf(self)669class TFLongformerModel:670 def __init__(self, *args, **kwargs):671 requires_tf(self)672 @classmethod673 def from_pretrained(self, *args, **kwargs):674 requires_tf(self)675class TFLongformerSelfAttention:676 def __init__(self, *args, **kwargs):677 requires_tf(self)678TF_LXMERT_PRETRAINED_MODEL_ARCHIVE_LIST = None679class TFLxmertForPreTraining:680 def __init__(self, *args, **kwargs):681 requires_tf(self)682class TFLxmertMainLayer:683 def __init__(self, *args, **kwargs):684 requires_tf(self)685class TFLxmertModel:686 def __init__(self, *args, **kwargs):687 requires_tf(self)688 @classmethod689 def from_pretrained(self, *args, **kwargs):690 requires_tf(self)691class TFLxmertPreTrainedModel:692 def __init__(self, *args, **kwargs):693 requires_tf(self)694 @classmethod695 def from_pretrained(self, *args, **kwargs):696 requires_tf(self)697class TFLxmertVisualFeatureEncoder:698 def __init__(self, *args, **kwargs):699 requires_tf(self)700class TFMarianModel:701 def __init__(self, *args, **kwargs):702 requires_tf(self)703 @classmethod704 def from_pretrained(self, *args, **kwargs):705 requires_tf(self)706class TFMarianMTModel:707 def __init__(self, *args, **kwargs):708 requires_tf(self)709 @classmethod710 def from_pretrained(self, *args, **kwargs):711 requires_tf(self)712class TFMBartForConditionalGeneration:713 def __init__(self, *args, **kwargs):714 requires_tf(self)715 @classmethod716 def from_pretrained(self, *args, **kwargs):717 requires_tf(self)718class TFMBartModel:719 def __init__(self, *args, **kwargs):720 requires_tf(self)721 @classmethod722 def from_pretrained(self, *args, **kwargs):723 requires_tf(self)724TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None725class TFMobileBertForMaskedLM:726 def __init__(self, *args, **kwargs):727 requires_tf(self)728 @classmethod729 def from_pretrained(self, *args, **kwargs):730 requires_tf(self)731class TFMobileBertForMultipleChoice:732 def __init__(self, *args, **kwargs):733 requires_tf(self)734 @classmethod735 def from_pretrained(self, *args, **kwargs):736 requires_tf(self)737class TFMobileBertForNextSentencePrediction:738 def __init__(self, *args, **kwargs):739 requires_tf(self)740class TFMobileBertForPreTraining:741 def __init__(self, *args, **kwargs):742 requires_tf(self)743class TFMobileBertForQuestionAnswering:744 def __init__(self, *args, **kwargs):745 requires_tf(self)746 @classmethod747 def from_pretrained(self, *args, **kwargs):748 requires_tf(self)749class TFMobileBertForSequenceClassification:750 def __init__(self, *args, **kwargs):751 requires_tf(self)752 @classmethod753 def from_pretrained(self, *args, **kwargs):754 requires_tf(self)755class TFMobileBertForTokenClassification:756 def __init__(self, *args, **kwargs):757 requires_tf(self)758 @classmethod759 def from_pretrained(self, *args, **kwargs):760 requires_tf(self)761class TFMobileBertMainLayer:762 def __init__(self, *args, **kwargs):763 requires_tf(self)764class TFMobileBertModel:765 def __init__(self, *args, **kwargs):766 requires_tf(self)767 @classmethod768 def from_pretrained(self, *args, **kwargs):769 requires_tf(self)770class TFMobileBertPreTrainedModel:771 def __init__(self, *args, **kwargs):772 requires_tf(self)773 @classmethod774 def from_pretrained(self, *args, **kwargs):775 requires_tf(self)776TF_MPNET_PRETRAINED_MODEL_ARCHIVE_LIST = None777class TFMPNetForMaskedLM:778 def __init__(self, *args, **kwargs):779 requires_tf(self)780 @classmethod781 def from_pretrained(self, *args, **kwargs):782 requires_tf(self)783class TFMPNetForMultipleChoice:784 def __init__(self, *args, **kwargs):785 requires_tf(self)786 @classmethod787 def from_pretrained(self, *args, **kwargs):788 requires_tf(self)789class TFMPNetForQuestionAnswering:790 def __init__(self, *args, **kwargs):791 requires_tf(self)792 @classmethod793 def from_pretrained(self, *args, **kwargs):794 requires_tf(self)795class TFMPNetForSequenceClassification:796 def __init__(self, *args, **kwargs):797 requires_tf(self)798 @classmethod799 def from_pretrained(self, *args, **kwargs):800 requires_tf(self)801class TFMPNetForTokenClassification:802 def __init__(self, *args, **kwargs):803 requires_tf(self)804 @classmethod805 def from_pretrained(self, *args, **kwargs):806 requires_tf(self)807class TFMPNetMainLayer:808 def __init__(self, *args, **kwargs):809 requires_tf(self)810class TFMPNetModel:811 def __init__(self, *args, **kwargs):812 requires_tf(self)813 @classmethod814 def from_pretrained(self, *args, **kwargs):815 requires_tf(self)816class TFMPNetPreTrainedModel:817 def __init__(self, *args, **kwargs):818 requires_tf(self)819 @classmethod820 def from_pretrained(self, *args, **kwargs):821 requires_tf(self)822class TFMT5EncoderModel:823 def __init__(self, *args, **kwargs):824 requires_tf(self)825 @classmethod826 def from_pretrained(self, *args, **kwargs):827 requires_tf(self)828class TFMT5ForConditionalGeneration:829 def __init__(self, *args, **kwargs):830 requires_tf(self)831 @classmethod832 def from_pretrained(self, *args, **kwargs):833 requires_tf(self)834class TFMT5Model:835 def __init__(self, *args, **kwargs):836 requires_tf(self)837 @classmethod838 def from_pretrained(self, *args, **kwargs):839 requires_tf(self)840TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST = None841class TFOpenAIGPTDoubleHeadsModel:842 def __init__(self, *args, **kwargs):843 requires_tf(self)844 @classmethod845 def from_pretrained(self, *args, **kwargs):846 requires_tf(self)847class TFOpenAIGPTForSequenceClassification:848 def __init__(self, *args, **kwargs):849 requires_tf(self)850 @classmethod851 def from_pretrained(self, *args, **kwargs):852 requires_tf(self)853class TFOpenAIGPTLMHeadModel:854 def __init__(self, *args, **kwargs):855 requires_tf(self)856 @classmethod857 def from_pretrained(self, *args, **kwargs):858 requires_tf(self)859class TFOpenAIGPTMainLayer:860 def __init__(self, *args, **kwargs):861 requires_tf(self)862class TFOpenAIGPTModel:863 def __init__(self, *args, **kwargs):864 requires_tf(self)865 @classmethod866 def from_pretrained(self, *args, **kwargs):867 requires_tf(self)868class TFOpenAIGPTPreTrainedModel:869 def __init__(self, *args, **kwargs):870 requires_tf(self)871 @classmethod872 def from_pretrained(self, *args, **kwargs):873 requires_tf(self)874class TFPegasusForConditionalGeneration:875 def __init__(self, *args, **kwargs):876 requires_tf(self)877 @classmethod878 def from_pretrained(self, *args, **kwargs):879 requires_tf(self)880class TFPegasusModel:881 def __init__(self, *args, **kwargs):882 requires_tf(self)883 @classmethod884 def from_pretrained(self, *args, **kwargs):885 requires_tf(self)886class TFRagModel:887 def __init__(self, *args, **kwargs):888 requires_tf(self)889 @classmethod890 def from_pretrained(self, *args, **kwargs):891 requires_tf(self)892class TFRagSequenceForGeneration:893 def __init__(self, *args, **kwargs):894 requires_tf(self)895class TFRagTokenForGeneration:896 def __init__(self, *args, **kwargs):897 requires_tf(self)898TF_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None899class TFRobertaForMaskedLM:900 def __init__(self, *args, **kwargs):901 requires_tf(self)902 @classmethod903 def from_pretrained(self, *args, **kwargs):904 requires_tf(self)905class TFRobertaForMultipleChoice:906 def __init__(self, *args, **kwargs):907 requires_tf(self)908 @classmethod909 def from_pretrained(self, *args, **kwargs):910 requires_tf(self)911class TFRobertaForQuestionAnswering:912 def __init__(self, *args, **kwargs):913 requires_tf(self)914 @classmethod915 def from_pretrained(self, *args, **kwargs):916 requires_tf(self)917class TFRobertaForSequenceClassification:918 def __init__(self, *args, **kwargs):919 requires_tf(self)920 @classmethod921 def from_pretrained(self, *args, **kwargs):922 requires_tf(self)923class TFRobertaForTokenClassification:924 def __init__(self, *args, **kwargs):925 requires_tf(self)926 @classmethod927 def from_pretrained(self, *args, **kwargs):928 requires_tf(self)929class TFRobertaMainLayer:930 def __init__(self, *args, **kwargs):931 requires_tf(self)932class TFRobertaModel:933 def __init__(self, *args, **kwargs):934 requires_tf(self)935 @classmethod936 def from_pretrained(self, *args, **kwargs):937 requires_tf(self)938class TFRobertaPreTrainedModel:939 def __init__(self, *args, **kwargs):940 requires_tf(self)941 @classmethod942 def from_pretrained(self, *args, **kwargs):943 requires_tf(self)944TF_T5_PRETRAINED_MODEL_ARCHIVE_LIST = None945class TFT5EncoderModel:946 def __init__(self, *args, **kwargs):947 requires_tf(self)948 @classmethod949 def from_pretrained(self, *args, **kwargs):950 requires_tf(self)951class TFT5ForConditionalGeneration:952 def __init__(self, *args, **kwargs):953 requires_tf(self)954 @classmethod955 def from_pretrained(self, *args, **kwargs):956 requires_tf(self)957class TFT5Model:958 def __init__(self, *args, **kwargs):959 requires_tf(self)960 @classmethod961 def from_pretrained(self, *args, **kwargs):962 requires_tf(self)963class TFT5PreTrainedModel:964 def __init__(self, *args, **kwargs):965 requires_tf(self)966 @classmethod967 def from_pretrained(self, *args, **kwargs):968 requires_tf(self)969TF_TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LIST = None970class TFAdaptiveEmbedding:971 def __init__(self, *args, **kwargs):972 requires_tf(self)973class TFTransfoXLForSequenceClassification:974 def __init__(self, *args, **kwargs):975 requires_tf(self)976 @classmethod977 def from_pretrained(self, *args, **kwargs):978 requires_tf(self)979class TFTransfoXLLMHeadModel:980 def __init__(self, *args, **kwargs):981 requires_tf(self)982 @classmethod983 def from_pretrained(self, *args, **kwargs):984 requires_tf(self)985class TFTransfoXLMainLayer:986 def __init__(self, *args, **kwargs):987 requires_tf(self)988class TFTransfoXLModel:989 def __init__(self, *args, **kwargs):990 requires_tf(self)991 @classmethod992 def from_pretrained(self, *args, **kwargs):993 requires_tf(self)994class TFTransfoXLPreTrainedModel:995 def __init__(self, *args, **kwargs):996 requires_tf(self)997 @classmethod998 def from_pretrained(self, *args, **kwargs):999 requires_tf(self)1000TF_XLM_PRETRAINED_MODEL_ARCHIVE_LIST = None1001class TFXLMForMultipleChoice:1002 def __init__(self, *args, **kwargs):1003 requires_tf(self)1004 @classmethod1005 def from_pretrained(self, *args, **kwargs):1006 requires_tf(self)1007class TFXLMForQuestionAnsweringSimple:1008 def __init__(self, *args, **kwargs):1009 requires_tf(self)1010 @classmethod1011 def from_pretrained(self, *args, **kwargs):1012 requires_tf(self)1013class TFXLMForSequenceClassification:1014 def __init__(self, *args, **kwargs):1015 requires_tf(self)1016 @classmethod1017 def from_pretrained(self, *args, **kwargs):1018 requires_tf(self)1019class TFXLMForTokenClassification:1020 def __init__(self, *args, **kwargs):1021 requires_tf(self)1022 @classmethod1023 def from_pretrained(self, *args, **kwargs):1024 requires_tf(self)1025class TFXLMMainLayer:1026 def __init__(self, *args, **kwargs):1027 requires_tf(self)1028class TFXLMModel:1029 def __init__(self, *args, **kwargs):1030 requires_tf(self)1031 @classmethod1032 def from_pretrained(self, *args, **kwargs):1033 requires_tf(self)1034class TFXLMPreTrainedModel:1035 def __init__(self, *args, **kwargs):1036 requires_tf(self)1037 @classmethod1038 def from_pretrained(self, *args, **kwargs):1039 requires_tf(self)1040class TFXLMWithLMHeadModel:1041 def __init__(self, *args, **kwargs):1042 requires_tf(self)1043 @classmethod1044 def from_pretrained(self, *args, **kwargs):1045 requires_tf(self)1046TF_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None1047class TFXLMRobertaForMaskedLM:1048 def __init__(self, *args, **kwargs):1049 requires_tf(self)1050 @classmethod1051 def from_pretrained(self, *args, **kwargs):1052 requires_tf(self)1053class TFXLMRobertaForMultipleChoice:1054 def __init__(self, *args, **kwargs):1055 requires_tf(self)1056 @classmethod1057 def from_pretrained(self, *args, **kwargs):1058 requires_tf(self)1059class TFXLMRobertaForQuestionAnswering:1060 def __init__(self, *args, **kwargs):1061 requires_tf(self)1062 @classmethod1063 def from_pretrained(self, *args, **kwargs):1064 requires_tf(self)1065class TFXLMRobertaForSequenceClassification:1066 def __init__(self, *args, **kwargs):1067 requires_tf(self)1068 @classmethod1069 def from_pretrained(self, *args, **kwargs):1070 requires_tf(self)1071class TFXLMRobertaForTokenClassification:1072 def __init__(self, *args, **kwargs):1073 requires_tf(self)1074 @classmethod1075 def from_pretrained(self, *args, **kwargs):1076 requires_tf(self)1077class TFXLMRobertaModel:1078 def __init__(self, *args, **kwargs):1079 requires_tf(self)1080 @classmethod1081 def from_pretrained(self, *args, **kwargs):1082 requires_tf(self)1083TF_XLNET_PRETRAINED_MODEL_ARCHIVE_LIST = None1084class TFXLNetForMultipleChoice:1085 def __init__(self, *args, **kwargs):1086 requires_tf(self)1087 @classmethod1088 def from_pretrained(self, *args, **kwargs):1089 requires_tf(self)1090class TFXLNetForQuestionAnsweringSimple:1091 def __init__(self, *args, **kwargs):1092 requires_tf(self)1093 @classmethod1094 def from_pretrained(self, *args, **kwargs):1095 requires_tf(self)1096class TFXLNetForSequenceClassification:1097 def __init__(self, *args, **kwargs):1098 requires_tf(self)1099 @classmethod1100 def from_pretrained(self, *args, **kwargs):1101 requires_tf(self)1102class TFXLNetForTokenClassification:1103 def __init__(self, *args, **kwargs):1104 requires_tf(self)1105 @classmethod1106 def from_pretrained(self, *args, **kwargs):1107 requires_tf(self)1108class TFXLNetLMHeadModel:1109 def __init__(self, *args, **kwargs):1110 requires_tf(self)1111 @classmethod1112 def from_pretrained(self, *args, **kwargs):1113 requires_tf(self)1114class TFXLNetMainLayer:1115 def __init__(self, *args, **kwargs):1116 requires_tf(self)1117class TFXLNetModel:1118 def __init__(self, *args, **kwargs):1119 requires_tf(self)1120 @classmethod1121 def from_pretrained(self, *args, **kwargs):1122 requires_tf(self)1123class TFXLNetPreTrainedModel:1124 def __init__(self, *args, **kwargs):1125 requires_tf(self)1126 @classmethod1127 def from_pretrained(self, *args, **kwargs):1128 requires_tf(self)1129class AdamWeightDecay:1130 def __init__(self, *args, **kwargs):1131 requires_tf(self)1132class GradientAccumulator:1133 def __init__(self, *args, **kwargs):1134 requires_tf(self)1135class WarmUp:1136 def __init__(self, *args, **kwargs):1137 requires_tf(self)1138def create_optimizer(*args, **kwargs):1139 requires_tf(create_optimizer)1140class TFTrainer:1141 def __init__(self, *args, **kwargs):...

Full Screen

Full Screen

dummy_flax_objects.py

Source:dummy_flax_objects.py Github

copy

Full Screen

1# This file is autogenerated by the command `make fix-copies`, do not edit.2from ..file_utils import requires_backends3class FlaxForcedBOSTokenLogitsProcessor:4 def __init__(self, *args, **kwargs):5 requires_backends(self, ["flax"])6 @classmethod7 def from_pretrained(cls, *args, **kwargs):8 requires_backends(cls, ["flax"])9class FlaxForcedEOSTokenLogitsProcessor:10 def __init__(self, *args, **kwargs):11 requires_backends(self, ["flax"])12 @classmethod13 def from_pretrained(cls, *args, **kwargs):14 requires_backends(cls, ["flax"])15class FlaxLogitsProcessor:16 def __init__(self, *args, **kwargs):17 requires_backends(self, ["flax"])18 @classmethod19 def from_pretrained(cls, *args, **kwargs):20 requires_backends(cls, ["flax"])21class FlaxLogitsProcessorList:22 def __init__(self, *args, **kwargs):23 requires_backends(self, ["flax"])24 @classmethod25 def from_pretrained(cls, *args, **kwargs):26 requires_backends(cls, ["flax"])27class FlaxLogitsWarper:28 def __init__(self, *args, **kwargs):29 requires_backends(self, ["flax"])30class FlaxMinLengthLogitsProcessor:31 def __init__(self, *args, **kwargs):32 requires_backends(self, ["flax"])33 @classmethod34 def from_pretrained(cls, *args, **kwargs):35 requires_backends(cls, ["flax"])36class FlaxTemperatureLogitsWarper:37 def __init__(self, *args, **kwargs):38 requires_backends(self, ["flax"])39class FlaxTopKLogitsWarper:40 def __init__(self, *args, **kwargs):41 requires_backends(self, ["flax"])42class FlaxTopPLogitsWarper:43 def __init__(self, *args, **kwargs):44 requires_backends(self, ["flax"])45class FlaxPreTrainedModel:46 def __init__(self, *args, **kwargs):47 requires_backends(self, ["flax"])48 @classmethod49 def from_pretrained(cls, *args, **kwargs):50 requires_backends(cls, ["flax"])51class FlaxAlbertForMaskedLM:52 def __init__(self, *args, **kwargs):53 requires_backends(self, ["flax"])54 @classmethod55 def from_pretrained(cls, *args, **kwargs):56 requires_backends(cls, ["flax"])57class FlaxAlbertForMultipleChoice:58 def __init__(self, *args, **kwargs):59 requires_backends(self, ["flax"])60 @classmethod61 def from_pretrained(cls, *args, **kwargs):62 requires_backends(cls, ["flax"])63class FlaxAlbertForPreTraining:64 def __init__(self, *args, **kwargs):65 requires_backends(self, ["flax"])66class FlaxAlbertForQuestionAnswering:67 def __init__(self, *args, **kwargs):68 requires_backends(self, ["flax"])69 @classmethod70 def from_pretrained(cls, *args, **kwargs):71 requires_backends(cls, ["flax"])72class FlaxAlbertForSequenceClassification:73 def __init__(self, *args, **kwargs):74 requires_backends(self, ["flax"])75 @classmethod76 def from_pretrained(cls, *args, **kwargs):77 requires_backends(cls, ["flax"])78class FlaxAlbertForTokenClassification:79 def __init__(self, *args, **kwargs):80 requires_backends(self, ["flax"])81 @classmethod82 def from_pretrained(cls, *args, **kwargs):83 requires_backends(cls, ["flax"])84class FlaxAlbertModel:85 def __init__(self, *args, **kwargs):86 requires_backends(self, ["flax"])87 @classmethod88 def from_pretrained(cls, *args, **kwargs):89 requires_backends(cls, ["flax"])90class FlaxAlbertPreTrainedModel:91 def __init__(self, *args, **kwargs):92 requires_backends(self, ["flax"])93 @classmethod94 def from_pretrained(cls, *args, **kwargs):95 requires_backends(cls, ["flax"])96FLAX_MODEL_FOR_CAUSAL_LM_MAPPING = None97FLAX_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING = None98FLAX_MODEL_FOR_MASKED_LM_MAPPING = None99FLAX_MODEL_FOR_MULTIPLE_CHOICE_MAPPING = None100FLAX_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING = None101FLAX_MODEL_FOR_PRETRAINING_MAPPING = None102FLAX_MODEL_FOR_QUESTION_ANSWERING_MAPPING = None103FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING = None104FLAX_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = None105FLAX_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING = None106FLAX_MODEL_MAPPING = None107class FlaxAutoModel:108 def __init__(self, *args, **kwargs):109 requires_backends(self, ["flax"])110 @classmethod111 def from_pretrained(cls, *args, **kwargs):112 requires_backends(cls, ["flax"])113class FlaxAutoModelForCausalLM:114 def __init__(self, *args, **kwargs):115 requires_backends(self, ["flax"])116 @classmethod117 def from_pretrained(cls, *args, **kwargs):118 requires_backends(cls, ["flax"])119class FlaxAutoModelForImageClassification:120 def __init__(self, *args, **kwargs):121 requires_backends(self, ["flax"])122 @classmethod123 def from_pretrained(cls, *args, **kwargs):124 requires_backends(cls, ["flax"])125class FlaxAutoModelForMaskedLM:126 def __init__(self, *args, **kwargs):127 requires_backends(self, ["flax"])128 @classmethod129 def from_pretrained(cls, *args, **kwargs):130 requires_backends(cls, ["flax"])131class FlaxAutoModelForMultipleChoice:132 def __init__(self, *args, **kwargs):133 requires_backends(self, ["flax"])134 @classmethod135 def from_pretrained(cls, *args, **kwargs):136 requires_backends(cls, ["flax"])137class FlaxAutoModelForNextSentencePrediction:138 def __init__(self, *args, **kwargs):139 requires_backends(self, ["flax"])140 @classmethod141 def from_pretrained(cls, *args, **kwargs):142 requires_backends(cls, ["flax"])143class FlaxAutoModelForPreTraining:144 def __init__(self, *args, **kwargs):145 requires_backends(self, ["flax"])146 @classmethod147 def from_pretrained(cls, *args, **kwargs):148 requires_backends(cls, ["flax"])149class FlaxAutoModelForQuestionAnswering:150 def __init__(self, *args, **kwargs):151 requires_backends(self, ["flax"])152 @classmethod153 def from_pretrained(cls, *args, **kwargs):154 requires_backends(cls, ["flax"])155class FlaxAutoModelForSeq2SeqLM:156 def __init__(self, *args, **kwargs):157 requires_backends(self, ["flax"])158 @classmethod159 def from_pretrained(cls, *args, **kwargs):160 requires_backends(cls, ["flax"])161class FlaxAutoModelForSequenceClassification:162 def __init__(self, *args, **kwargs):163 requires_backends(self, ["flax"])164 @classmethod165 def from_pretrained(cls, *args, **kwargs):166 requires_backends(cls, ["flax"])167class FlaxAutoModelForTokenClassification:168 def __init__(self, *args, **kwargs):169 requires_backends(self, ["flax"])170 @classmethod171 def from_pretrained(cls, *args, **kwargs):172 requires_backends(cls, ["flax"])173class FlaxBartForConditionalGeneration:174 def __init__(self, *args, **kwargs):175 requires_backends(self, ["flax"])176 @classmethod177 def from_pretrained(cls, *args, **kwargs):178 requires_backends(cls, ["flax"])179class FlaxBartForQuestionAnswering:180 def __init__(self, *args, **kwargs):181 requires_backends(self, ["flax"])182 @classmethod183 def from_pretrained(cls, *args, **kwargs):184 requires_backends(cls, ["flax"])185class FlaxBartForSequenceClassification:186 def __init__(self, *args, **kwargs):187 requires_backends(self, ["flax"])188 @classmethod189 def from_pretrained(cls, *args, **kwargs):190 requires_backends(cls, ["flax"])191class FlaxBartModel:192 def __init__(self, *args, **kwargs):193 requires_backends(self, ["flax"])194 @classmethod195 def from_pretrained(cls, *args, **kwargs):196 requires_backends(cls, ["flax"])197class FlaxBartPreTrainedModel:198 def __init__(self, *args, **kwargs):199 requires_backends(self, ["flax"])200 @classmethod201 def from_pretrained(cls, *args, **kwargs):202 requires_backends(cls, ["flax"])203class FlaxBertForMaskedLM:204 def __init__(self, *args, **kwargs):205 requires_backends(self, ["flax"])206 @classmethod207 def from_pretrained(cls, *args, **kwargs):208 requires_backends(cls, ["flax"])209class FlaxBertForMultipleChoice:210 def __init__(self, *args, **kwargs):211 requires_backends(self, ["flax"])212 @classmethod213 def from_pretrained(cls, *args, **kwargs):214 requires_backends(cls, ["flax"])215class FlaxBertForNextSentencePrediction:216 def __init__(self, *args, **kwargs):217 requires_backends(self, ["flax"])218class FlaxBertForPreTraining:219 def __init__(self, *args, **kwargs):220 requires_backends(self, ["flax"])221class FlaxBertForQuestionAnswering:222 def __init__(self, *args, **kwargs):223 requires_backends(self, ["flax"])224 @classmethod225 def from_pretrained(cls, *args, **kwargs):226 requires_backends(cls, ["flax"])227class FlaxBertForSequenceClassification:228 def __init__(self, *args, **kwargs):229 requires_backends(self, ["flax"])230 @classmethod231 def from_pretrained(cls, *args, **kwargs):232 requires_backends(cls, ["flax"])233class FlaxBertForTokenClassification:234 def __init__(self, *args, **kwargs):235 requires_backends(self, ["flax"])236 @classmethod237 def from_pretrained(cls, *args, **kwargs):238 requires_backends(cls, ["flax"])239class FlaxBertModel:240 def __init__(self, *args, **kwargs):241 requires_backends(self, ["flax"])242 @classmethod243 def from_pretrained(cls, *args, **kwargs):244 requires_backends(cls, ["flax"])245class FlaxBertPreTrainedModel:246 def __init__(self, *args, **kwargs):247 requires_backends(self, ["flax"])248 @classmethod249 def from_pretrained(cls, *args, **kwargs):250 requires_backends(cls, ["flax"])251class FlaxBigBirdForMaskedLM:252 def __init__(self, *args, **kwargs):253 requires_backends(self, ["flax"])254 @classmethod255 def from_pretrained(cls, *args, **kwargs):256 requires_backends(cls, ["flax"])257class FlaxBigBirdForMultipleChoice:258 def __init__(self, *args, **kwargs):259 requires_backends(self, ["flax"])260 @classmethod261 def from_pretrained(cls, *args, **kwargs):262 requires_backends(cls, ["flax"])263class FlaxBigBirdForPreTraining:264 def __init__(self, *args, **kwargs):265 requires_backends(self, ["flax"])266class FlaxBigBirdForQuestionAnswering:267 def __init__(self, *args, **kwargs):268 requires_backends(self, ["flax"])269 @classmethod270 def from_pretrained(cls, *args, **kwargs):271 requires_backends(cls, ["flax"])272class FlaxBigBirdForSequenceClassification:273 def __init__(self, *args, **kwargs):274 requires_backends(self, ["flax"])275 @classmethod276 def from_pretrained(cls, *args, **kwargs):277 requires_backends(cls, ["flax"])278class FlaxBigBirdForTokenClassification:279 def __init__(self, *args, **kwargs):280 requires_backends(self, ["flax"])281 @classmethod282 def from_pretrained(cls, *args, **kwargs):283 requires_backends(cls, ["flax"])284class FlaxBigBirdModel:285 def __init__(self, *args, **kwargs):286 requires_backends(self, ["flax"])287 @classmethod288 def from_pretrained(cls, *args, **kwargs):289 requires_backends(cls, ["flax"])290class FlaxBigBirdPreTrainedModel:291 def __init__(self, *args, **kwargs):292 requires_backends(self, ["flax"])293 @classmethod294 def from_pretrained(cls, *args, **kwargs):295 requires_backends(cls, ["flax"])296class FlaxCLIPModel:297 def __init__(self, *args, **kwargs):298 requires_backends(self, ["flax"])299 @classmethod300 def from_pretrained(cls, *args, **kwargs):301 requires_backends(cls, ["flax"])302class FlaxCLIPPreTrainedModel:303 def __init__(self, *args, **kwargs):304 requires_backends(self, ["flax"])305 @classmethod306 def from_pretrained(cls, *args, **kwargs):307 requires_backends(cls, ["flax"])308class FlaxCLIPTextModel:309 def __init__(self, *args, **kwargs):310 requires_backends(self, ["flax"])311 @classmethod312 def from_pretrained(cls, *args, **kwargs):313 requires_backends(cls, ["flax"])314class FlaxCLIPTextPreTrainedModel:315 def __init__(self, *args, **kwargs):316 requires_backends(self, ["flax"])317 @classmethod318 def from_pretrained(cls, *args, **kwargs):319 requires_backends(cls, ["flax"])320class FlaxCLIPVisionModel:321 def __init__(self, *args, **kwargs):322 requires_backends(self, ["flax"])323 @classmethod324 def from_pretrained(cls, *args, **kwargs):325 requires_backends(cls, ["flax"])326class FlaxCLIPVisionPreTrainedModel:327 def __init__(self, *args, **kwargs):328 requires_backends(self, ["flax"])329 @classmethod330 def from_pretrained(cls, *args, **kwargs):331 requires_backends(cls, ["flax"])332class FlaxDistilBertForMaskedLM:333 def __init__(self, *args, **kwargs):334 requires_backends(self, ["flax"])335 @classmethod336 def from_pretrained(cls, *args, **kwargs):337 requires_backends(cls, ["flax"])338class FlaxDistilBertForMultipleChoice:339 def __init__(self, *args, **kwargs):340 requires_backends(self, ["flax"])341 @classmethod342 def from_pretrained(cls, *args, **kwargs):343 requires_backends(cls, ["flax"])344class FlaxDistilBertForQuestionAnswering:345 def __init__(self, *args, **kwargs):346 requires_backends(self, ["flax"])347 @classmethod348 def from_pretrained(cls, *args, **kwargs):349 requires_backends(cls, ["flax"])350class FlaxDistilBertForSequenceClassification:351 def __init__(self, *args, **kwargs):352 requires_backends(self, ["flax"])353 @classmethod354 def from_pretrained(cls, *args, **kwargs):355 requires_backends(cls, ["flax"])356class FlaxDistilBertForTokenClassification:357 def __init__(self, *args, **kwargs):358 requires_backends(self, ["flax"])359 @classmethod360 def from_pretrained(cls, *args, **kwargs):361 requires_backends(cls, ["flax"])362class FlaxDistilBertModel:363 def __init__(self, *args, **kwargs):364 requires_backends(self, ["flax"])365 @classmethod366 def from_pretrained(cls, *args, **kwargs):367 requires_backends(cls, ["flax"])368class FlaxDistilBertPreTrainedModel:369 def __init__(self, *args, **kwargs):370 requires_backends(self, ["flax"])371 @classmethod372 def from_pretrained(cls, *args, **kwargs):373 requires_backends(cls, ["flax"])374class FlaxElectraForMaskedLM:375 def __init__(self, *args, **kwargs):376 requires_backends(self, ["flax"])377 @classmethod378 def from_pretrained(cls, *args, **kwargs):379 requires_backends(cls, ["flax"])380class FlaxElectraForMultipleChoice:381 def __init__(self, *args, **kwargs):382 requires_backends(self, ["flax"])383 @classmethod384 def from_pretrained(cls, *args, **kwargs):385 requires_backends(cls, ["flax"])386class FlaxElectraForPreTraining:387 def __init__(self, *args, **kwargs):388 requires_backends(self, ["flax"])389class FlaxElectraForQuestionAnswering:390 def __init__(self, *args, **kwargs):391 requires_backends(self, ["flax"])392 @classmethod393 def from_pretrained(cls, *args, **kwargs):394 requires_backends(cls, ["flax"])395class FlaxElectraForSequenceClassification:396 def __init__(self, *args, **kwargs):397 requires_backends(self, ["flax"])398 @classmethod399 def from_pretrained(cls, *args, **kwargs):400 requires_backends(cls, ["flax"])401class FlaxElectraForTokenClassification:402 def __init__(self, *args, **kwargs):403 requires_backends(self, ["flax"])404 @classmethod405 def from_pretrained(cls, *args, **kwargs):406 requires_backends(cls, ["flax"])407class FlaxElectraModel:408 def __init__(self, *args, **kwargs):409 requires_backends(self, ["flax"])410 @classmethod411 def from_pretrained(cls, *args, **kwargs):412 requires_backends(cls, ["flax"])413class FlaxElectraPreTrainedModel:414 def __init__(self, *args, **kwargs):415 requires_backends(self, ["flax"])416 @classmethod417 def from_pretrained(cls, *args, **kwargs):418 requires_backends(cls, ["flax"])419class FlaxEncoderDecoderModel:420 def __init__(self, *args, **kwargs):421 requires_backends(self, ["flax"])422 @classmethod423 def from_pretrained(cls, *args, **kwargs):424 requires_backends(cls, ["flax"])425class FlaxGPT2LMHeadModel:426 def __init__(self, *args, **kwargs):427 requires_backends(self, ["flax"])428 @classmethod429 def from_pretrained(cls, *args, **kwargs):430 requires_backends(cls, ["flax"])431class FlaxGPT2Model:432 def __init__(self, *args, **kwargs):433 requires_backends(self, ["flax"])434 @classmethod435 def from_pretrained(cls, *args, **kwargs):436 requires_backends(cls, ["flax"])437class FlaxGPT2PreTrainedModel:438 def __init__(self, *args, **kwargs):439 requires_backends(self, ["flax"])440 @classmethod441 def from_pretrained(cls, *args, **kwargs):442 requires_backends(cls, ["flax"])443class FlaxGPTNeoForCausalLM:444 def __init__(self, *args, **kwargs):445 requires_backends(self, ["flax"])446 @classmethod447 def from_pretrained(cls, *args, **kwargs):448 requires_backends(cls, ["flax"])449class FlaxGPTNeoModel:450 def __init__(self, *args, **kwargs):451 requires_backends(self, ["flax"])452 @classmethod453 def from_pretrained(cls, *args, **kwargs):454 requires_backends(cls, ["flax"])455class FlaxGPTNeoPreTrainedModel:456 def __init__(self, *args, **kwargs):457 requires_backends(self, ["flax"])458 @classmethod459 def from_pretrained(cls, *args, **kwargs):460 requires_backends(cls, ["flax"])461class FlaxMarianModel:462 def __init__(self, *args, **kwargs):463 requires_backends(self, ["flax"])464 @classmethod465 def from_pretrained(cls, *args, **kwargs):466 requires_backends(cls, ["flax"])467class FlaxMarianMTModel:468 def __init__(self, *args, **kwargs):469 requires_backends(self, ["flax"])470 @classmethod471 def from_pretrained(cls, *args, **kwargs):472 requires_backends(cls, ["flax"])473class FlaxMarianPreTrainedModel:474 def __init__(self, *args, **kwargs):475 requires_backends(self, ["flax"])476 @classmethod477 def from_pretrained(cls, *args, **kwargs):478 requires_backends(cls, ["flax"])479class FlaxMBartForConditionalGeneration:480 def __init__(self, *args, **kwargs):481 requires_backends(self, ["flax"])482 @classmethod483 def from_pretrained(cls, *args, **kwargs):484 requires_backends(cls, ["flax"])485class FlaxMBartForQuestionAnswering:486 def __init__(self, *args, **kwargs):487 requires_backends(self, ["flax"])488 @classmethod489 def from_pretrained(cls, *args, **kwargs):490 requires_backends(cls, ["flax"])491class FlaxMBartForSequenceClassification:492 def __init__(self, *args, **kwargs):493 requires_backends(self, ["flax"])494 @classmethod495 def from_pretrained(cls, *args, **kwargs):496 requires_backends(cls, ["flax"])497class FlaxMBartModel:498 def __init__(self, *args, **kwargs):499 requires_backends(self, ["flax"])500 @classmethod501 def from_pretrained(cls, *args, **kwargs):502 requires_backends(cls, ["flax"])503class FlaxMBartPreTrainedModel:504 def __init__(self, *args, **kwargs):505 requires_backends(self, ["flax"])506 @classmethod507 def from_pretrained(cls, *args, **kwargs):508 requires_backends(cls, ["flax"])509class FlaxMT5ForConditionalGeneration:510 def __init__(self, *args, **kwargs):511 requires_backends(self, ["flax"])512 @classmethod513 def from_pretrained(cls, *args, **kwargs):514 requires_backends(cls, ["flax"])515class FlaxMT5Model:516 def __init__(self, *args, **kwargs):517 requires_backends(self, ["flax"])518 @classmethod519 def from_pretrained(cls, *args, **kwargs):520 requires_backends(cls, ["flax"])521class FlaxRobertaForMaskedLM:522 def __init__(self, *args, **kwargs):523 requires_backends(self, ["flax"])524 @classmethod525 def from_pretrained(cls, *args, **kwargs):526 requires_backends(cls, ["flax"])527class FlaxRobertaForMultipleChoice:528 def __init__(self, *args, **kwargs):529 requires_backends(self, ["flax"])530 @classmethod531 def from_pretrained(cls, *args, **kwargs):532 requires_backends(cls, ["flax"])533class FlaxRobertaForQuestionAnswering:534 def __init__(self, *args, **kwargs):535 requires_backends(self, ["flax"])536 @classmethod537 def from_pretrained(cls, *args, **kwargs):538 requires_backends(cls, ["flax"])539class FlaxRobertaForSequenceClassification:540 def __init__(self, *args, **kwargs):541 requires_backends(self, ["flax"])542 @classmethod543 def from_pretrained(cls, *args, **kwargs):544 requires_backends(cls, ["flax"])545class FlaxRobertaForTokenClassification:546 def __init__(self, *args, **kwargs):547 requires_backends(self, ["flax"])548 @classmethod549 def from_pretrained(cls, *args, **kwargs):550 requires_backends(cls, ["flax"])551class FlaxRobertaModel:552 def __init__(self, *args, **kwargs):553 requires_backends(self, ["flax"])554 @classmethod555 def from_pretrained(cls, *args, **kwargs):556 requires_backends(cls, ["flax"])557class FlaxRobertaPreTrainedModel:558 def __init__(self, *args, **kwargs):559 requires_backends(self, ["flax"])560 @classmethod561 def from_pretrained(cls, *args, **kwargs):562 requires_backends(cls, ["flax"])563class FlaxT5ForConditionalGeneration:564 def __init__(self, *args, **kwargs):565 requires_backends(self, ["flax"])566 @classmethod567 def from_pretrained(cls, *args, **kwargs):568 requires_backends(cls, ["flax"])569class FlaxT5Model:570 def __init__(self, *args, **kwargs):571 requires_backends(self, ["flax"])572 @classmethod573 def from_pretrained(cls, *args, **kwargs):574 requires_backends(cls, ["flax"])575class FlaxT5PreTrainedModel:576 def __init__(self, *args, **kwargs):577 requires_backends(self, ["flax"])578 @classmethod579 def from_pretrained(cls, *args, **kwargs):580 requires_backends(cls, ["flax"])581class FlaxViTForImageClassification:582 def __init__(self, *args, **kwargs):583 requires_backends(self, ["flax"])584class FlaxViTModel:585 def __init__(self, *args, **kwargs):586 requires_backends(self, ["flax"])587 @classmethod588 def from_pretrained(cls, *args, **kwargs):589 requires_backends(cls, ["flax"])590class FlaxViTPreTrainedModel:591 def __init__(self, *args, **kwargs):592 requires_backends(self, ["flax"])593 @classmethod594 def from_pretrained(cls, *args, **kwargs):595 requires_backends(cls, ["flax"])596class FlaxWav2Vec2ForCTC:597 def __init__(self, *args, **kwargs):598 requires_backends(self, ["flax"])599class FlaxWav2Vec2ForPreTraining:600 def __init__(self, *args, **kwargs):601 requires_backends(self, ["flax"])602class FlaxWav2Vec2Model:603 def __init__(self, *args, **kwargs):604 requires_backends(self, ["flax"])605 @classmethod606 def from_pretrained(cls, *args, **kwargs):607 requires_backends(cls, ["flax"])608class FlaxWav2Vec2PreTrainedModel:609 def __init__(self, *args, **kwargs):610 requires_backends(self, ["flax"])611 @classmethod612 def from_pretrained(cls, *args, **kwargs):...

Full Screen

Full Screen

test_manifest.py

Source:test_manifest.py Github

copy

Full Screen

1# This Source Code Form is subject to the terms of the Mozilla Public2# License, v. 2.0. If a copy of the MPL was not distributed with this3# file, You can obtain one at http://mozilla.org/MPL/2.0/.4import unittest5from StringIO import StringIO6from cuddlefish.manifest import scan_module7class Extra:8 def failUnlessKeysAre(self, d, keys):9 self.failUnlessEqual(sorted(d.keys()), sorted(keys))10class Require(unittest.TestCase, Extra):11 def scan(self, text):12 lines = StringIO(text).readlines()13 requires, problems, locations = scan_module("fake.js", lines)14 self.failUnlessEqual(problems, False)15 return requires16 def scan_locations(self, text):17 lines = StringIO(text).readlines()18 requires, problems, locations = scan_module("fake.js", lines)19 self.failUnlessEqual(problems, False)20 return requires, locations21 def test_modules(self):22 mod = """var foo = require('one');"""23 requires = self.scan(mod)24 self.failUnlessKeysAre(requires, ["one"])25 mod = """var foo = require(\"one\");"""26 requires = self.scan(mod)27 self.failUnlessKeysAre(requires, ["one"])28 mod = """var foo=require( 'one' ) ; """29 requires = self.scan(mod)30 self.failUnlessKeysAre(requires, ["one"])31 mod = """var foo = require('o'+'ne'); // tricky, denied"""32 requires = self.scan(mod)33 self.failUnlessKeysAre(requires, [])34 mod = """require('one').immediately.do().stuff();"""35 requires, locations = self.scan_locations(mod)36 self.failUnlessKeysAre(requires, ["one"])37 self.failUnlessEqual(locations, {"one": 1})38 # these forms are commented out, and thus ignored39 mod = """// var foo = require('one');"""40 requires = self.scan(mod)41 self.failUnlessKeysAre(requires, [])42 mod = """/* var foo = require('one');"""43 requires = self.scan(mod)44 self.failUnlessKeysAre(requires, [])45 mod = """ * var foo = require('one');"""46 requires = self.scan(mod)47 self.failUnlessKeysAre(requires, [])48 mod = """ ' var foo = require('one');"""49 requires = self.scan(mod)50 self.failUnlessKeysAre(requires, ["one"])51 mod = """ \" var foo = require('one');"""52 requires = self.scan(mod)53 self.failUnlessKeysAre(requires, ["one"])54 # multiple requires55 mod = """const foo = require('one');56 const foo = require('two');"""57 requires, locations = self.scan_locations(mod)58 self.failUnlessKeysAre(requires, ["one", "two"])59 self.failUnlessEqual(locations["one"], 1)60 self.failUnlessEqual(locations["two"], 2)61 mod = """const foo = require('repeated');62 const bar = require('repeated');63 const baz = require('repeated');"""64 requires, locations = self.scan_locations(mod)65 self.failUnlessKeysAre(requires, ["repeated"])66 self.failUnlessEqual(locations["repeated"], 1) # first occurrence67 mod = """const foo = require('one'); const foo = require('two');"""68 requires = self.scan(mod)69 self.failUnlessKeysAre(requires, ["one", "two"])70 # define calls71 mod = """define('one', ['two', 'numbers/three'], function(t, th) {});"""72 requires = self.scan(mod)73 self.failUnlessKeysAre(requires, ["two", "numbers/three"])74 mod = """define(75 ['odd',76 "numbers/four"], function() {});"""77 requires = self.scan(mod)78 self.failUnlessKeysAre(requires, ["odd", "numbers/four"])79 mod = """define(function(require, exports, module) {80 var a = require("some/module/a"),81 b = require('b/v1');82 exports.a = a;83 //This is a fakeout: require('bad');84 /* And another var bad = require('bad2'); */85 require('foo').goFoo();86 });"""87 requires = self.scan(mod)88 self.failUnlessKeysAre(requires, ["some/module/a", "b/v1", "foo"])89 mod = """define (90 "foo",91 ["bar"], function (bar) {92 var me = require("me");93 }94 )"""95 requires = self.scan(mod)96 self.failUnlessKeysAre(requires, ["bar", "me"])97 mod = """define(['se' + 'ven', 'eight', nine], function () {});"""98 requires = self.scan(mod)99 self.failUnlessKeysAre(requires, ["eight"])100 # async require calls101 mod = """require(['one'], function(one) {var o = require("one");});"""102 requires = self.scan(mod)103 self.failUnlessKeysAre(requires, ["one"])104 mod = """require([ 'one' ], function(one) {var t = require("two");});"""105 requires = self.scan(mod)106 self.failUnlessKeysAre(requires, ["one", "two"])107 mod = """require ( ['two', 'numbers/three'], function(t, th) {});"""108 requires = self.scan(mod)109 self.failUnlessKeysAre(requires, ["two", "numbers/three"])110 mod = """require (111 ["bar", "fa" + 'ke' ], function (bar) {112 var me = require("me");113 // require("bad").doBad();114 }115 )"""116 requires = self.scan(mod)117 self.failUnlessKeysAre(requires, ["bar", "me"])118def scan2(text, fn="fake.js"):119 stderr = StringIO()120 lines = StringIO(text).readlines()121 requires, problems, locations = scan_module(fn, lines, stderr)122 stderr.seek(0)123 return requires, problems, stderr.readlines()124class Chrome(unittest.TestCase, Extra):125 def test_ignore_loader(self):126 # we specifically ignore the loader itself127 mod = """let {Cc,Ci} = require('chrome');"""128 requires, problems, err = scan2(mod, "blah/cuddlefish.js")129 self.failUnlessKeysAre(requires, ["chrome"])130 self.failUnlessEqual(problems, False)131 self.failUnlessEqual(err, [])132 def test_chrome(self):133 mod = """let {Cc,Ci} = require('chrome');"""134 requires, problems, err = scan2(mod)135 self.failUnlessKeysAre(requires, ["chrome"])136 self.failUnlessEqual(problems, False)137 self.failUnlessEqual(err, [])138 mod = """var foo = require('foo');139 let {Cc,Ci} = require('chrome');"""140 requires, problems, err = scan2(mod)141 self.failUnlessKeysAre(requires, ["foo", "chrome"])142 self.failUnlessEqual(problems, False)143 self.failUnlessEqual(err, [])144 mod = """let c = require('chrome');"""145 requires, problems, err = scan2(mod)146 self.failUnlessKeysAre(requires, ["chrome"])147 self.failUnlessEqual(problems, False)148 self.failUnlessEqual(err, [])149 mod = """var foo = require('foo');150 let c = require('chrome');"""151 requires, problems, err = scan2(mod)152 self.failUnlessKeysAre(requires, ["foo", "chrome"])153 self.failUnlessEqual(problems, False)154 self.failUnlessEqual(err, [])155 def test_chrome_components(self):156 # Bug 663541: tolerate "Components" if you're marked with157 # require("chrome"), to avoid requiring module authors to rewrite a158 # lot of code. Once bug 636145 is fixed, such code will break. To fix159 # it, add {Components}=require("chrome"), but that won't work until160 # after 636145 is fixed.161 mod = """require("chrome");162 var ios = Components.classes['@mozilla.org/network/io-service;1'];"""163 requires, problems, err = scan2(mod)164 self.failUnlessKeysAre(requires, ["chrome"])165 self.failUnlessEqual((problems, err), (False, []))166 def test_not_chrome(self):167 # from bug 596595168 mod = r'soughtLines: new RegExp("^\\s*(\\[[0-9 .]*\\])?\\s*\\(\\((EE|WW)\\)|.* [Cc]hipsets?: \\)|\\s*Backtrace")'169 requires, problems, err = scan2(mod)170 self.failUnlessKeysAre(requires, [])171 self.failUnlessEqual((problems,err), (False, []))172 def test_not_chrome2(self):173 # from bug 655788174 mod = r"var foo = 'some stuff Cr';"175 requires, problems, err = scan2(mod)176 self.failUnlessKeysAre(requires, [])177 self.failUnlessEqual((problems,err), (False, []))178class BadChrome(unittest.TestCase, Extra):179 def test_bad_alias(self):180 # using Components.* gets you an error, with a message that teaches181 # you the correct approach.182 mod = """let Cc = Components.classes;183 let Cu = Components.utils;184 """185 requires, problems, err = scan2(mod)186 self.failUnlessKeysAre(requires, [])187 self.failUnlessEqual(problems, True)188 self.failUnlessEqual(err[1], "The following lines from file fake.js:\n")189 self.failUnlessEqual(err[2], " 1: let Cc = Components.classes;\n")190 self.failUnlessEqual(err[3], " 2: let Cu = Components.utils;\n")191 self.failUnlessEqual(err[4], "use 'Components' to access chrome authority. To do so, you need to add a\n")192 self.failUnlessEqual(err[5], "line somewhat like the following:\n")193 self.failUnlessEqual(err[7], ' const {Cc,Cu} = require("chrome");\n')194 self.failUnlessEqual(err[9], "Then you can use 'Components' as well as any shortcuts to its properties\n")195 def test_bad_misc(self):196 # If it looks like you're using something that doesn't have an alias,197 # the warning also suggests a better way.198 mod = """if (Components.isSuccessCode(foo))199 """200 requires, problems, err = scan2(mod)201 self.failUnlessKeysAre(requires, [])202 self.failUnlessEqual(problems, True)203 self.failUnlessEqual(err[1], "The following lines from file fake.js:\n")204 self.failUnlessEqual(err[2], " 1: if (Components.isSuccessCode(foo))\n")205 self.failUnlessEqual(err[3], "use 'Components' to access chrome authority. To do so, you need to add a\n")206 self.failUnlessEqual(err[4], "line somewhat like the following:\n")207 self.failUnlessEqual(err[6], ' const {components} = require("chrome");\n')208 self.failUnlessEqual(err[8], "Then you can use 'Components' as well as any shortcuts to its properties\n")209if __name__ == '__main__':...

Full Screen

Full Screen

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.

LambdaTest Learning Hubs:

YouTube

You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.

Run Slash automation tests on LambdaTest cloud grid

Perform automation testing on 3000+ real desktop and mobile devices online.

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

Next-Gen App & Browser Testing Cloud

Was this article helpful?

Helpful

NotHelpful