Fix init import_structure sorting (#20477)
* Fix init import_structure sorting * Fix rebase
This commit is contained in:
@@ -569,10 +569,10 @@ else:
|
|||||||
_import_structure["models.m2m_100"].append("M2M100Tokenizer")
|
_import_structure["models.m2m_100"].append("M2M100Tokenizer")
|
||||||
_import_structure["models.marian"].append("MarianTokenizer")
|
_import_structure["models.marian"].append("MarianTokenizer")
|
||||||
_import_structure["models.mbart"].append("MBartTokenizer")
|
_import_structure["models.mbart"].append("MBartTokenizer")
|
||||||
_import_structure["models.nllb"].append("NllbTokenizer")
|
|
||||||
_import_structure["models.mbart50"].append("MBart50Tokenizer")
|
_import_structure["models.mbart50"].append("MBart50Tokenizer")
|
||||||
_import_structure["models.mluke"].append("MLukeTokenizer")
|
_import_structure["models.mluke"].append("MLukeTokenizer")
|
||||||
_import_structure["models.mt5"].append("MT5Tokenizer")
|
_import_structure["models.mt5"].append("MT5Tokenizer")
|
||||||
|
_import_structure["models.nllb"].append("NllbTokenizer")
|
||||||
_import_structure["models.pegasus"].append("PegasusTokenizer")
|
_import_structure["models.pegasus"].append("PegasusTokenizer")
|
||||||
_import_structure["models.plbart"].append("PLBartTokenizer")
|
_import_structure["models.plbart"].append("PLBartTokenizer")
|
||||||
_import_structure["models.reformer"].append("ReformerTokenizer")
|
_import_structure["models.reformer"].append("ReformerTokenizer")
|
||||||
@@ -722,14 +722,14 @@ else:
|
|||||||
_import_structure["image_utils"] = ["ImageFeatureExtractionMixin"]
|
_import_structure["image_utils"] = ["ImageFeatureExtractionMixin"]
|
||||||
_import_structure["models.beit"].extend(["BeitFeatureExtractor", "BeitImageProcessor"])
|
_import_structure["models.beit"].extend(["BeitFeatureExtractor", "BeitImageProcessor"])
|
||||||
_import_structure["models.clip"].extend(["CLIPFeatureExtractor", "CLIPImageProcessor"])
|
_import_structure["models.clip"].extend(["CLIPFeatureExtractor", "CLIPImageProcessor"])
|
||||||
|
_import_structure["models.conditional_detr"].append("ConditionalDetrFeatureExtractor")
|
||||||
_import_structure["models.convnext"].extend(["ConvNextFeatureExtractor", "ConvNextImageProcessor"])
|
_import_structure["models.convnext"].extend(["ConvNextFeatureExtractor", "ConvNextImageProcessor"])
|
||||||
_import_structure["models.deformable_detr"].append("DeformableDetrFeatureExtractor")
|
_import_structure["models.deformable_detr"].append("DeformableDetrFeatureExtractor")
|
||||||
_import_structure["models.deit"].extend(["DeiTFeatureExtractor", "DeiTImageProcessor"])
|
_import_structure["models.deit"].extend(["DeiTFeatureExtractor", "DeiTImageProcessor"])
|
||||||
_import_structure["models.detr"].append("DetrFeatureExtractor")
|
_import_structure["models.detr"].append("DetrFeatureExtractor")
|
||||||
_import_structure["models.conditional_detr"].append("ConditionalDetrFeatureExtractor")
|
|
||||||
_import_structure["models.donut"].extend(["DonutFeatureExtractor", "DonutImageProcessor"])
|
_import_structure["models.donut"].extend(["DonutFeatureExtractor", "DonutImageProcessor"])
|
||||||
_import_structure["models.dpt"].extend(["DPTFeatureExtractor", "DPTImageProcessor"])
|
_import_structure["models.dpt"].extend(["DPTFeatureExtractor", "DPTImageProcessor"])
|
||||||
_import_structure["models.flava"].extend(["FlavaFeatureExtractor", "FlavaProcessor", "FlavaImageProcessor"])
|
_import_structure["models.flava"].extend(["FlavaFeatureExtractor", "FlavaImageProcessor", "FlavaProcessor"])
|
||||||
_import_structure["models.glpn"].extend(["GLPNFeatureExtractor", "GLPNImageProcessor"])
|
_import_structure["models.glpn"].extend(["GLPNFeatureExtractor", "GLPNImageProcessor"])
|
||||||
_import_structure["models.imagegpt"].extend(["ImageGPTFeatureExtractor", "ImageGPTImageProcessor"])
|
_import_structure["models.imagegpt"].extend(["ImageGPTFeatureExtractor", "ImageGPTImageProcessor"])
|
||||||
_import_structure["models.layoutlmv2"].extend(["LayoutLMv2FeatureExtractor", "LayoutLMv2ImageProcessor"])
|
_import_structure["models.layoutlmv2"].extend(["LayoutLMv2FeatureExtractor", "LayoutLMv2ImageProcessor"])
|
||||||
@@ -819,70 +819,44 @@ else:
|
|||||||
"TextDatasetForNextSentencePrediction",
|
"TextDatasetForNextSentencePrediction",
|
||||||
]
|
]
|
||||||
_import_structure["deepspeed"] = []
|
_import_structure["deepspeed"] = []
|
||||||
_import_structure["generation_utils"] = []
|
|
||||||
_import_structure["generation"].extend(
|
_import_structure["generation"].extend(
|
||||||
[
|
[
|
||||||
"Constraint",
|
|
||||||
"ConstraintListState",
|
|
||||||
"DisjunctiveConstraint",
|
|
||||||
"PhrasalConstraint",
|
|
||||||
"BeamScorer",
|
"BeamScorer",
|
||||||
"BeamSearchScorer",
|
"BeamSearchScorer",
|
||||||
"ConstrainedBeamSearchScorer",
|
"ConstrainedBeamSearchScorer",
|
||||||
|
"Constraint",
|
||||||
|
"ConstraintListState",
|
||||||
|
"DisjunctiveConstraint",
|
||||||
"ForcedBOSTokenLogitsProcessor",
|
"ForcedBOSTokenLogitsProcessor",
|
||||||
"ForcedEOSTokenLogitsProcessor",
|
"ForcedEOSTokenLogitsProcessor",
|
||||||
|
"GenerationMixin",
|
||||||
"HammingDiversityLogitsProcessor",
|
"HammingDiversityLogitsProcessor",
|
||||||
"InfNanRemoveLogitsProcessor",
|
"InfNanRemoveLogitsProcessor",
|
||||||
"LogitsProcessor",
|
"LogitsProcessor",
|
||||||
"LogitsProcessorList",
|
"LogitsProcessorList",
|
||||||
"LogitsWarper",
|
"LogitsWarper",
|
||||||
|
"MaxLengthCriteria",
|
||||||
|
"MaxTimeCriteria",
|
||||||
"MinLengthLogitsProcessor",
|
"MinLengthLogitsProcessor",
|
||||||
"NoBadWordsLogitsProcessor",
|
"NoBadWordsLogitsProcessor",
|
||||||
"NoRepeatNGramLogitsProcessor",
|
"NoRepeatNGramLogitsProcessor",
|
||||||
|
"PhrasalConstraint",
|
||||||
"PrefixConstrainedLogitsProcessor",
|
"PrefixConstrainedLogitsProcessor",
|
||||||
"RepetitionPenaltyLogitsProcessor",
|
"RepetitionPenaltyLogitsProcessor",
|
||||||
|
"StoppingCriteria",
|
||||||
|
"StoppingCriteriaList",
|
||||||
"TemperatureLogitsWarper",
|
"TemperatureLogitsWarper",
|
||||||
"TopKLogitsWarper",
|
"TopKLogitsWarper",
|
||||||
"TopPLogitsWarper",
|
"TopPLogitsWarper",
|
||||||
"TypicalLogitsWarper",
|
"TypicalLogitsWarper",
|
||||||
"MaxLengthCriteria",
|
|
||||||
"MaxTimeCriteria",
|
|
||||||
"StoppingCriteria",
|
|
||||||
"StoppingCriteriaList",
|
|
||||||
"GenerationMixin",
|
|
||||||
"top_k_top_p_filtering",
|
"top_k_top_p_filtering",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
_import_structure["generation_utils"] = []
|
||||||
_import_structure["modeling_outputs"] = []
|
_import_structure["modeling_outputs"] = []
|
||||||
_import_structure["modeling_utils"] = ["PreTrainedModel"]
|
_import_structure["modeling_utils"] = ["PreTrainedModel"]
|
||||||
|
|
||||||
# PyTorch models structure
|
# PyTorch models structure
|
||||||
|
|
||||||
_import_structure["models.roc_bert"].extend(
|
|
||||||
[
|
|
||||||
"ROC_BERT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
|
||||||
"RoCBertForMaskedLM",
|
|
||||||
"RoCBertForCausalLM",
|
|
||||||
"RoCBertForMultipleChoice",
|
|
||||||
"RoCBertForQuestionAnswering",
|
|
||||||
"RoCBertForSequenceClassification",
|
|
||||||
"RoCBertForTokenClassification",
|
|
||||||
"RoCBertLayer",
|
|
||||||
"RoCBertModel",
|
|
||||||
"RoCBertForPreTraining",
|
|
||||||
"RoCBertPreTrainedModel",
|
|
||||||
"load_tf_weights_in_roc_bert",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
|
|
||||||
_import_structure["models.time_series_transformer"].extend(
|
|
||||||
[
|
|
||||||
"TIME_SERIES_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LIST",
|
|
||||||
"TimeSeriesTransformerForPrediction",
|
|
||||||
"TimeSeriesTransformerModel",
|
|
||||||
"TimeSeriesTransformerPreTrainedModel",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
_import_structure["models.albert"].extend(
|
_import_structure["models.albert"].extend(
|
||||||
[
|
[
|
||||||
"ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -897,12 +871,13 @@ else:
|
|||||||
"load_tf_weights_in_albert",
|
"load_tf_weights_in_albert",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
_import_structure["models.audio_spectrogram_transformer"].extend(
|
_import_structure["models.audio_spectrogram_transformer"].extend(
|
||||||
[
|
[
|
||||||
"AUDIO_SPECTROGRAM_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"AUDIO_SPECTROGRAM_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"ASTForAudioClassification",
|
||||||
"ASTModel",
|
"ASTModel",
|
||||||
"ASTPreTrainedModel",
|
"ASTPreTrainedModel",
|
||||||
"ASTForAudioClassification",
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.auto"].extend(
|
_import_structure["models.auto"].extend(
|
||||||
@@ -913,8 +888,8 @@ else:
|
|||||||
"MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPING",
|
"MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPING",
|
||||||
"MODEL_FOR_CAUSAL_LM_MAPPING",
|
"MODEL_FOR_CAUSAL_LM_MAPPING",
|
||||||
"MODEL_FOR_CTC_MAPPING",
|
"MODEL_FOR_CTC_MAPPING",
|
||||||
"MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING",
|
|
||||||
"MODEL_FOR_DEPTH_ESTIMATION_MAPPING",
|
"MODEL_FOR_DEPTH_ESTIMATION_MAPPING",
|
||||||
|
"MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING",
|
||||||
"MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING",
|
"MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING",
|
||||||
"MODEL_FOR_IMAGE_SEGMENTATION_MAPPING",
|
"MODEL_FOR_IMAGE_SEGMENTATION_MAPPING",
|
||||||
"MODEL_FOR_INSTANCE_SEGMENTATION_MAPPING",
|
"MODEL_FOR_INSTANCE_SEGMENTATION_MAPPING",
|
||||||
@@ -934,18 +909,18 @@ else:
|
|||||||
"MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING",
|
"MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING",
|
||||||
"MODEL_FOR_VISION_2_SEQ_MAPPING",
|
"MODEL_FOR_VISION_2_SEQ_MAPPING",
|
||||||
"MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPING",
|
"MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPING",
|
||||||
|
"MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPING",
|
||||||
"MODEL_MAPPING",
|
"MODEL_MAPPING",
|
||||||
"MODEL_WITH_LM_HEAD_MAPPING",
|
"MODEL_WITH_LM_HEAD_MAPPING",
|
||||||
"MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPING",
|
|
||||||
"AutoModel",
|
|
||||||
"AutoBackbone",
|
"AutoBackbone",
|
||||||
|
"AutoModel",
|
||||||
"AutoModelForAudioClassification",
|
"AutoModelForAudioClassification",
|
||||||
"AutoModelForAudioFrameClassification",
|
"AutoModelForAudioFrameClassification",
|
||||||
"AutoModelForAudioXVector",
|
"AutoModelForAudioXVector",
|
||||||
"AutoModelForCausalLM",
|
"AutoModelForCausalLM",
|
||||||
"AutoModelForCTC",
|
"AutoModelForCTC",
|
||||||
"AutoModelForDocumentQuestionAnswering",
|
|
||||||
"AutoModelForDepthEstimation",
|
"AutoModelForDepthEstimation",
|
||||||
|
"AutoModelForDocumentQuestionAnswering",
|
||||||
"AutoModelForImageClassification",
|
"AutoModelForImageClassification",
|
||||||
"AutoModelForImageSegmentation",
|
"AutoModelForImageSegmentation",
|
||||||
"AutoModelForInstanceSegmentation",
|
"AutoModelForInstanceSegmentation",
|
||||||
@@ -965,8 +940,8 @@ else:
|
|||||||
"AutoModelForVideoClassification",
|
"AutoModelForVideoClassification",
|
||||||
"AutoModelForVision2Seq",
|
"AutoModelForVision2Seq",
|
||||||
"AutoModelForVisualQuestionAnswering",
|
"AutoModelForVisualQuestionAnswering",
|
||||||
"AutoModelWithLMHead",
|
|
||||||
"AutoModelForZeroShotObjectDetection",
|
"AutoModelForZeroShotObjectDetection",
|
||||||
|
"AutoModelWithLMHead",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.bart"].extend(
|
_import_structure["models.bart"].extend(
|
||||||
@@ -981,17 +956,6 @@ else:
|
|||||||
"PretrainedBartModel",
|
"PretrainedBartModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.mvp"].extend(
|
|
||||||
[
|
|
||||||
"MVP_PRETRAINED_MODEL_ARCHIVE_LIST",
|
|
||||||
"MvpForCausalLM",
|
|
||||||
"MvpForConditionalGeneration",
|
|
||||||
"MvpForQuestionAnswering",
|
|
||||||
"MvpForSequenceClassification",
|
|
||||||
"MvpModel",
|
|
||||||
"MvpPreTrainedModel",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
_import_structure["models.beit"].extend(
|
_import_structure["models.beit"].extend(
|
||||||
[
|
[
|
||||||
"BEIT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"BEIT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -1054,17 +1018,6 @@ else:
|
|||||||
"BigBirdPegasusPreTrainedModel",
|
"BigBirdPegasusPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.bloom"].extend(
|
|
||||||
[
|
|
||||||
"BLOOM_PRETRAINED_MODEL_ARCHIVE_LIST",
|
|
||||||
"BloomForCausalLM",
|
|
||||||
"BloomModel",
|
|
||||||
"BloomPreTrainedModel",
|
|
||||||
"BloomForSequenceClassification",
|
|
||||||
"BloomForTokenClassification",
|
|
||||||
"BloomForQuestionAnswering",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
_import_structure["models.blenderbot"].extend(
|
_import_structure["models.blenderbot"].extend(
|
||||||
[
|
[
|
||||||
"BLENDERBOT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"BLENDERBOT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -1083,6 +1036,17 @@ else:
|
|||||||
"BlenderbotSmallPreTrainedModel",
|
"BlenderbotSmallPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
_import_structure["models.bloom"].extend(
|
||||||
|
[
|
||||||
|
"BLOOM_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"BloomForCausalLM",
|
||||||
|
"BloomForQuestionAnswering",
|
||||||
|
"BloomForSequenceClassification",
|
||||||
|
"BloomForTokenClassification",
|
||||||
|
"BloomModel",
|
||||||
|
"BloomPreTrainedModel",
|
||||||
|
]
|
||||||
|
)
|
||||||
_import_structure["models.camembert"].extend(
|
_import_structure["models.camembert"].extend(
|
||||||
[
|
[
|
||||||
"CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -1123,20 +1087,19 @@ else:
|
|||||||
_import_structure["models.clipseg"].extend(
|
_import_structure["models.clipseg"].extend(
|
||||||
[
|
[
|
||||||
"CLIPSEG_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"CLIPSEG_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"CLIPSegForImageSegmentation",
|
||||||
"CLIPSegModel",
|
"CLIPSegModel",
|
||||||
"CLIPSegPreTrainedModel",
|
"CLIPSegPreTrainedModel",
|
||||||
"CLIPSegTextModel",
|
"CLIPSegTextModel",
|
||||||
"CLIPSegVisionModel",
|
"CLIPSegVisionModel",
|
||||||
"CLIPSegForImageSegmentation",
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.x_clip"].extend(
|
_import_structure["models.codegen"].extend(
|
||||||
[
|
[
|
||||||
"XCLIP_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"CODEGEN_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"XCLIPModel",
|
"CodeGenForCausalLM",
|
||||||
"XCLIPPreTrainedModel",
|
"CodeGenModel",
|
||||||
"XCLIPTextModel",
|
"CodeGenPreTrainedModel",
|
||||||
"XCLIPVisionModel",
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.convbert"].extend(
|
_import_structure["models.convbert"].extend(
|
||||||
@@ -1245,6 +1208,14 @@ else:
|
|||||||
"DeiTPreTrainedModel",
|
"DeiTPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
_import_structure["models.dinat"].extend(
|
||||||
|
[
|
||||||
|
"DINAT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"DinatForImageClassification",
|
||||||
|
"DinatModel",
|
||||||
|
"DinatPreTrainedModel",
|
||||||
|
]
|
||||||
|
)
|
||||||
_import_structure["models.distilbert"].extend(
|
_import_structure["models.distilbert"].extend(
|
||||||
[
|
[
|
||||||
"DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -1257,14 +1228,6 @@ else:
|
|||||||
"DistilBertPreTrainedModel",
|
"DistilBertPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.dinat"].extend(
|
|
||||||
[
|
|
||||||
"DINAT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
|
||||||
"DinatForImageClassification",
|
|
||||||
"DinatModel",
|
|
||||||
"DinatPreTrainedModel",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
_import_structure["models.donut"].extend(
|
_import_structure["models.donut"].extend(
|
||||||
[
|
[
|
||||||
"DONUT_SWIN_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"DONUT_SWIN_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -1347,8 +1310,8 @@ else:
|
|||||||
"FlaubertForSequenceClassification",
|
"FlaubertForSequenceClassification",
|
||||||
"FlaubertForTokenClassification",
|
"FlaubertForTokenClassification",
|
||||||
"FlaubertModel",
|
"FlaubertModel",
|
||||||
"FlaubertWithLMHeadModel",
|
|
||||||
"FlaubertPreTrainedModel",
|
"FlaubertPreTrainedModel",
|
||||||
|
"FlaubertWithLMHeadModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.flava"].extend(
|
_import_structure["models.flava"].extend(
|
||||||
@@ -1461,14 +1424,6 @@ else:
|
|||||||
"GroupViTVisionModel",
|
"GroupViTVisionModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.codegen"].extend(
|
|
||||||
[
|
|
||||||
"CODEGEN_PRETRAINED_MODEL_ARCHIVE_LIST",
|
|
||||||
"CodeGenForCausalLM",
|
|
||||||
"CodeGenModel",
|
|
||||||
"CodeGenPreTrainedModel",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
_import_structure["models.hubert"].extend(
|
_import_structure["models.hubert"].extend(
|
||||||
[
|
[
|
||||||
"HUBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"HUBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -1505,17 +1460,17 @@ else:
|
|||||||
"JUKEBOX_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"JUKEBOX_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"JukeboxModel",
|
"JukeboxModel",
|
||||||
"JukeboxPreTrainedModel",
|
"JukeboxPreTrainedModel",
|
||||||
"JukeboxVQVAE",
|
|
||||||
"JukeboxPrior",
|
"JukeboxPrior",
|
||||||
|
"JukeboxVQVAE",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.layoutlm"].extend(
|
_import_structure["models.layoutlm"].extend(
|
||||||
[
|
[
|
||||||
"LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"LayoutLMForMaskedLM",
|
"LayoutLMForMaskedLM",
|
||||||
|
"LayoutLMForQuestionAnswering",
|
||||||
"LayoutLMForSequenceClassification",
|
"LayoutLMForSequenceClassification",
|
||||||
"LayoutLMForTokenClassification",
|
"LayoutLMForTokenClassification",
|
||||||
"LayoutLMForQuestionAnswering",
|
|
||||||
"LayoutLMModel",
|
"LayoutLMModel",
|
||||||
"LayoutLMPreTrainedModel",
|
"LayoutLMPreTrainedModel",
|
||||||
]
|
]
|
||||||
@@ -1559,6 +1514,16 @@ else:
|
|||||||
"LevitPreTrainedModel",
|
"LevitPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
_import_structure["models.lilt"].extend(
|
||||||
|
[
|
||||||
|
"LILT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"LiltForQuestionAnswering",
|
||||||
|
"LiltForSequenceClassification",
|
||||||
|
"LiltForTokenClassification",
|
||||||
|
"LiltModel",
|
||||||
|
"LiltPreTrainedModel",
|
||||||
|
]
|
||||||
|
)
|
||||||
_import_structure["models.longformer"].extend(
|
_import_structure["models.longformer"].extend(
|
||||||
[
|
[
|
||||||
"LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -1587,11 +1552,11 @@ else:
|
|||||||
"LukeForEntityClassification",
|
"LukeForEntityClassification",
|
||||||
"LukeForEntityPairClassification",
|
"LukeForEntityPairClassification",
|
||||||
"LukeForEntitySpanClassification",
|
"LukeForEntitySpanClassification",
|
||||||
|
"LukeForMaskedLM",
|
||||||
"LukeForMultipleChoice",
|
"LukeForMultipleChoice",
|
||||||
"LukeForQuestionAnswering",
|
"LukeForQuestionAnswering",
|
||||||
"LukeForSequenceClassification",
|
"LukeForSequenceClassification",
|
||||||
"LukeForTokenClassification",
|
"LukeForTokenClassification",
|
||||||
"LukeForMaskedLM",
|
|
||||||
"LukeModel",
|
"LukeModel",
|
||||||
"LukePreTrainedModel",
|
"LukePreTrainedModel",
|
||||||
]
|
]
|
||||||
@@ -1616,15 +1581,6 @@ else:
|
|||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.marian"].extend(["MarianForCausalLM", "MarianModel", "MarianMTModel"])
|
_import_structure["models.marian"].extend(["MarianForCausalLM", "MarianModel", "MarianMTModel"])
|
||||||
_import_structure["models.maskformer"].extend(
|
|
||||||
[
|
|
||||||
"MASKFORMER_PRETRAINED_MODEL_ARCHIVE_LIST",
|
|
||||||
"MaskFormerForInstanceSegmentation",
|
|
||||||
"MaskFormerModel",
|
|
||||||
"MaskFormerPreTrainedModel",
|
|
||||||
"MaskFormerSwinBackbone",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
_import_structure["models.markuplm"].extend(
|
_import_structure["models.markuplm"].extend(
|
||||||
[
|
[
|
||||||
"MARKUPLM_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"MARKUPLM_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -1635,6 +1591,15 @@ else:
|
|||||||
"MarkupLMPreTrainedModel",
|
"MarkupLMPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
_import_structure["models.maskformer"].extend(
|
||||||
|
[
|
||||||
|
"MASKFORMER_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"MaskFormerForInstanceSegmentation",
|
||||||
|
"MaskFormerModel",
|
||||||
|
"MaskFormerPreTrainedModel",
|
||||||
|
"MaskFormerSwinBackbone",
|
||||||
|
]
|
||||||
|
)
|
||||||
_import_structure["models.mbart"].extend(
|
_import_structure["models.mbart"].extend(
|
||||||
[
|
[
|
||||||
"MBartForCausalLM",
|
"MBartForCausalLM",
|
||||||
@@ -1727,6 +1692,17 @@ else:
|
|||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.mt5"].extend(["MT5EncoderModel", "MT5ForConditionalGeneration", "MT5Model"])
|
_import_structure["models.mt5"].extend(["MT5EncoderModel", "MT5ForConditionalGeneration", "MT5Model"])
|
||||||
|
_import_structure["models.mvp"].extend(
|
||||||
|
[
|
||||||
|
"MVP_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"MvpForCausalLM",
|
||||||
|
"MvpForConditionalGeneration",
|
||||||
|
"MvpForQuestionAnswering",
|
||||||
|
"MvpForSequenceClassification",
|
||||||
|
"MvpModel",
|
||||||
|
"MvpPreTrainedModel",
|
||||||
|
]
|
||||||
|
)
|
||||||
_import_structure["models.nat"].extend(
|
_import_structure["models.nat"].extend(
|
||||||
[
|
[
|
||||||
"NAT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"NAT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -1739,9 +1715,9 @@ else:
|
|||||||
[
|
[
|
||||||
"NEZHA_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"NEZHA_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"NezhaForMaskedLM",
|
"NezhaForMaskedLM",
|
||||||
"NezhaForPreTraining",
|
|
||||||
"NezhaForNextSentencePrediction",
|
|
||||||
"NezhaForMultipleChoice",
|
"NezhaForMultipleChoice",
|
||||||
|
"NezhaForNextSentencePrediction",
|
||||||
|
"NezhaForPreTraining",
|
||||||
"NezhaForQuestionAnswering",
|
"NezhaForQuestionAnswering",
|
||||||
"NezhaForSequenceClassification",
|
"NezhaForSequenceClassification",
|
||||||
"NezhaForTokenClassification",
|
"NezhaForTokenClassification",
|
||||||
@@ -1777,20 +1753,20 @@ else:
|
|||||||
[
|
[
|
||||||
"OPT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"OPT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"OPTForCausalLM",
|
"OPTForCausalLM",
|
||||||
|
"OPTForQuestionAnswering",
|
||||||
|
"OPTForSequenceClassification",
|
||||||
"OPTModel",
|
"OPTModel",
|
||||||
"OPTPreTrainedModel",
|
"OPTPreTrainedModel",
|
||||||
"OPTForSequenceClassification",
|
|
||||||
"OPTForQuestionAnswering",
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.owlvit"].extend(
|
_import_structure["models.owlvit"].extend(
|
||||||
[
|
[
|
||||||
"OWLVIT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"OWLVIT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"OwlViTForObjectDetection",
|
||||||
"OwlViTModel",
|
"OwlViTModel",
|
||||||
"OwlViTPreTrainedModel",
|
"OwlViTPreTrainedModel",
|
||||||
"OwlViTTextModel",
|
"OwlViTTextModel",
|
||||||
"OwlViTVisionModel",
|
"OwlViTVisionModel",
|
||||||
"OwlViTForObjectDetection",
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.pegasus"].extend(
|
_import_structure["models.pegasus"].extend(
|
||||||
@@ -1919,10 +1895,10 @@ else:
|
|||||||
_import_structure["models.resnet"].extend(
|
_import_structure["models.resnet"].extend(
|
||||||
[
|
[
|
||||||
"RESNET_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"RESNET_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"ResNetBackbone",
|
||||||
"ResNetForImageClassification",
|
"ResNetForImageClassification",
|
||||||
"ResNetModel",
|
"ResNetModel",
|
||||||
"ResNetPreTrainedModel",
|
"ResNetPreTrainedModel",
|
||||||
"ResNetBackbone",
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.retribert"].extend(
|
_import_structure["models.retribert"].extend(
|
||||||
@@ -1941,14 +1917,20 @@ else:
|
|||||||
"RobertaPreTrainedModel",
|
"RobertaPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.lilt"].extend(
|
_import_structure["models.roc_bert"].extend(
|
||||||
[
|
[
|
||||||
"LILT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"ROC_BERT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"LiltForQuestionAnswering",
|
"RoCBertForCausalLM",
|
||||||
"LiltForSequenceClassification",
|
"RoCBertForMaskedLM",
|
||||||
"LiltForTokenClassification",
|
"RoCBertForMultipleChoice",
|
||||||
"LiltModel",
|
"RoCBertForPreTraining",
|
||||||
"LiltPreTrainedModel",
|
"RoCBertForQuestionAnswering",
|
||||||
|
"RoCBertForSequenceClassification",
|
||||||
|
"RoCBertForTokenClassification",
|
||||||
|
"RoCBertLayer",
|
||||||
|
"RoCBertModel",
|
||||||
|
"RoCBertPreTrainedModel",
|
||||||
|
"load_tf_weights_in_roc_bert",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.roformer"].extend(
|
_import_structure["models.roformer"].extend(
|
||||||
@@ -2004,14 +1986,6 @@ else:
|
|||||||
"Speech2TextPreTrainedModel",
|
"Speech2TextPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.whisper"].extend(
|
|
||||||
[
|
|
||||||
"WHISPER_PRETRAINED_MODEL_ARCHIVE_LIST",
|
|
||||||
"WhisperForConditionalGeneration",
|
|
||||||
"WhisperModel",
|
|
||||||
"WhisperPreTrainedModel",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
_import_structure["models.speech_to_text_2"].extend(["Speech2Text2ForCausalLM", "Speech2Text2PreTrainedModel"])
|
_import_structure["models.speech_to_text_2"].extend(["Speech2Text2ForCausalLM", "Speech2Text2PreTrainedModel"])
|
||||||
_import_structure["models.splinter"].extend(
|
_import_structure["models.splinter"].extend(
|
||||||
[
|
[
|
||||||
@@ -2054,15 +2028,15 @@ else:
|
|||||||
"Swinv2PreTrainedModel",
|
"Swinv2PreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.tapas"].extend(
|
_import_structure["models.switch_transformers"].extend(
|
||||||
[
|
[
|
||||||
"TAPAS_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"SWITCH_TRANSFORMERS_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"TapasForMaskedLM",
|
"SwitchTransformersEncoderModel",
|
||||||
"TapasForQuestionAnswering",
|
"SwitchTransformersForConditionalGeneration",
|
||||||
"TapasForSequenceClassification",
|
"SwitchTransformersModel",
|
||||||
"TapasModel",
|
"SwitchTransformersPreTrainedModel",
|
||||||
"TapasPreTrainedModel",
|
"SwitchTransformersSparseMLP",
|
||||||
"load_tf_weights_in_tapas",
|
"SwitchTransformersTop1Router",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.t5"].extend(
|
_import_structure["models.t5"].extend(
|
||||||
@@ -2075,15 +2049,23 @@ else:
|
|||||||
"load_tf_weights_in_t5",
|
"load_tf_weights_in_t5",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.switch_transformers"].extend(
|
_import_structure["models.tapas"].extend(
|
||||||
[
|
[
|
||||||
"SWITCH_TRANSFORMERS_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"TAPAS_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"SwitchTransformersEncoderModel",
|
"TapasForMaskedLM",
|
||||||
"SwitchTransformersForConditionalGeneration",
|
"TapasForQuestionAnswering",
|
||||||
"SwitchTransformersModel",
|
"TapasForSequenceClassification",
|
||||||
"SwitchTransformersPreTrainedModel",
|
"TapasModel",
|
||||||
"SwitchTransformersTop1Router",
|
"TapasPreTrainedModel",
|
||||||
"SwitchTransformersSparseMLP",
|
"load_tf_weights_in_tapas",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
_import_structure["models.time_series_transformer"].extend(
|
||||||
|
[
|
||||||
|
"TIME_SERIES_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"TimeSeriesTransformerForPrediction",
|
||||||
|
"TimeSeriesTransformerModel",
|
||||||
|
"TimeSeriesTransformerPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.trajectory_transformer"].extend(
|
_import_structure["models.trajectory_transformer"].extend(
|
||||||
@@ -2137,14 +2119,23 @@ else:
|
|||||||
"VanPreTrainedModel",
|
"VanPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
_import_structure["models.videomae"].extend(
|
||||||
|
[
|
||||||
|
"VIDEOMAE_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"VideoMAEForPreTraining",
|
||||||
|
"VideoMAEForVideoClassification",
|
||||||
|
"VideoMAEModel",
|
||||||
|
"VideoMAEPreTrainedModel",
|
||||||
|
]
|
||||||
|
)
|
||||||
_import_structure["models.vilt"].extend(
|
_import_structure["models.vilt"].extend(
|
||||||
[
|
[
|
||||||
"VILT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"VILT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"ViltForImageAndTextRetrieval",
|
"ViltForImageAndTextRetrieval",
|
||||||
"ViltForImagesAndTextClassification",
|
"ViltForImagesAndTextClassification",
|
||||||
"ViltForTokenClassification",
|
|
||||||
"ViltForMaskedLM",
|
"ViltForMaskedLM",
|
||||||
"ViltForQuestionAnswering",
|
"ViltForQuestionAnswering",
|
||||||
|
"ViltForTokenClassification",
|
||||||
"ViltLayer",
|
"ViltLayer",
|
||||||
"ViltModel",
|
"ViltModel",
|
||||||
"ViltPreTrainedModel",
|
"ViltPreTrainedModel",
|
||||||
@@ -2186,20 +2177,11 @@ else:
|
|||||||
_import_structure["models.vit_msn"].extend(
|
_import_structure["models.vit_msn"].extend(
|
||||||
[
|
[
|
||||||
"VIT_MSN_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"VIT_MSN_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"ViTMSNModel",
|
|
||||||
"ViTMSNForImageClassification",
|
"ViTMSNForImageClassification",
|
||||||
|
"ViTMSNModel",
|
||||||
"ViTMSNPreTrainedModel",
|
"ViTMSNPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.videomae"].extend(
|
|
||||||
[
|
|
||||||
"VIDEOMAE_PRETRAINED_MODEL_ARCHIVE_LIST",
|
|
||||||
"VideoMAEForPreTraining",
|
|
||||||
"VideoMAEModel",
|
|
||||||
"VideoMAEPreTrainedModel",
|
|
||||||
"VideoMAEForVideoClassification",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
_import_structure["models.wav2vec2"].extend(
|
_import_structure["models.wav2vec2"].extend(
|
||||||
[
|
[
|
||||||
"WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -2236,6 +2218,23 @@ else:
|
|||||||
"WavLMPreTrainedModel",
|
"WavLMPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
_import_structure["models.whisper"].extend(
|
||||||
|
[
|
||||||
|
"WHISPER_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"WhisperForConditionalGeneration",
|
||||||
|
"WhisperModel",
|
||||||
|
"WhisperPreTrainedModel",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
_import_structure["models.x_clip"].extend(
|
||||||
|
[
|
||||||
|
"XCLIP_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
|
"XCLIPModel",
|
||||||
|
"XCLIPPreTrainedModel",
|
||||||
|
"XCLIPTextModel",
|
||||||
|
"XCLIPVisionModel",
|
||||||
|
]
|
||||||
|
)
|
||||||
_import_structure["models.xglm"].extend(
|
_import_structure["models.xglm"].extend(
|
||||||
[
|
[
|
||||||
"XGLM_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"XGLM_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
@@ -2358,11 +2357,11 @@ else:
|
|||||||
_import_structure["activations_tf"] = []
|
_import_structure["activations_tf"] = []
|
||||||
_import_structure["benchmark.benchmark_args_tf"] = ["TensorFlowBenchmarkArguments"]
|
_import_structure["benchmark.benchmark_args_tf"] = ["TensorFlowBenchmarkArguments"]
|
||||||
_import_structure["benchmark.benchmark_tf"] = ["TensorFlowBenchmark"]
|
_import_structure["benchmark.benchmark_tf"] = ["TensorFlowBenchmark"]
|
||||||
_import_structure["generation_tf_utils"] = []
|
|
||||||
_import_structure["generation"].extend(
|
_import_structure["generation"].extend(
|
||||||
[
|
[
|
||||||
"TFForcedBOSTokenLogitsProcessor",
|
"TFForcedBOSTokenLogitsProcessor",
|
||||||
"TFForcedEOSTokenLogitsProcessor",
|
"TFForcedEOSTokenLogitsProcessor",
|
||||||
|
"TFGenerationMixin",
|
||||||
"TFLogitsProcessor",
|
"TFLogitsProcessor",
|
||||||
"TFLogitsProcessorList",
|
"TFLogitsProcessorList",
|
||||||
"TFLogitsWarper",
|
"TFLogitsWarper",
|
||||||
@@ -2373,10 +2372,10 @@ else:
|
|||||||
"TFTemperatureLogitsWarper",
|
"TFTemperatureLogitsWarper",
|
||||||
"TFTopKLogitsWarper",
|
"TFTopKLogitsWarper",
|
||||||
"TFTopPLogitsWarper",
|
"TFTopPLogitsWarper",
|
||||||
"TFGenerationMixin",
|
|
||||||
"tf_top_k_top_p_filtering",
|
"tf_top_k_top_p_filtering",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
_import_structure["generation_tf_utils"] = []
|
||||||
_import_structure["keras_callbacks"] = ["KerasMetricCallback", "PushToHubCallback"]
|
_import_structure["keras_callbacks"] = ["KerasMetricCallback", "PushToHubCallback"]
|
||||||
_import_structure["modeling_tf_outputs"] = []
|
_import_structure["modeling_tf_outputs"] = []
|
||||||
_import_structure["modeling_tf_utils"] = [
|
_import_structure["modeling_tf_utils"] = [
|
||||||
@@ -2403,13 +2402,13 @@ else:
|
|||||||
_import_structure["models.auto"].extend(
|
_import_structure["models.auto"].extend(
|
||||||
[
|
[
|
||||||
"TF_MODEL_FOR_CAUSAL_LM_MAPPING",
|
"TF_MODEL_FOR_CAUSAL_LM_MAPPING",
|
||||||
|
"TF_MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING",
|
||||||
"TF_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING",
|
"TF_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING",
|
||||||
"TF_MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING",
|
"TF_MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING",
|
||||||
"TF_MODEL_FOR_MASKED_LM_MAPPING",
|
"TF_MODEL_FOR_MASKED_LM_MAPPING",
|
||||||
"TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING",
|
"TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING",
|
||||||
"TF_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING",
|
"TF_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING",
|
||||||
"TF_MODEL_FOR_PRETRAINING_MAPPING",
|
"TF_MODEL_FOR_PRETRAINING_MAPPING",
|
||||||
"TF_MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING",
|
|
||||||
"TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING",
|
"TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING",
|
||||||
"TF_MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING",
|
"TF_MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING",
|
||||||
"TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING",
|
"TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING",
|
||||||
@@ -2422,12 +2421,12 @@ else:
|
|||||||
"TF_MODEL_WITH_LM_HEAD_MAPPING",
|
"TF_MODEL_WITH_LM_HEAD_MAPPING",
|
||||||
"TFAutoModel",
|
"TFAutoModel",
|
||||||
"TFAutoModelForCausalLM",
|
"TFAutoModelForCausalLM",
|
||||||
|
"TFAutoModelForDocumentQuestionAnswering",
|
||||||
"TFAutoModelForImageClassification",
|
"TFAutoModelForImageClassification",
|
||||||
"TFAutoModelForMaskedLM",
|
"TFAutoModelForMaskedLM",
|
||||||
"TFAutoModelForMultipleChoice",
|
"TFAutoModelForMultipleChoice",
|
||||||
"TFAutoModelForNextSentencePrediction",
|
"TFAutoModelForNextSentencePrediction",
|
||||||
"TFAutoModelForPreTraining",
|
"TFAutoModelForPreTraining",
|
||||||
"TFAutoModelForDocumentQuestionAnswering",
|
|
||||||
"TFAutoModelForQuestionAnswering",
|
"TFAutoModelForQuestionAnswering",
|
||||||
"TFAutoModelForSemanticSegmentation",
|
"TFAutoModelForSemanticSegmentation",
|
||||||
"TFAutoModelForSeq2SeqLM",
|
"TFAutoModelForSeq2SeqLM",
|
||||||
@@ -2679,8 +2678,8 @@ else:
|
|||||||
[
|
[
|
||||||
"TF_LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"TF_LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"TFLayoutLMForMaskedLM",
|
"TFLayoutLMForMaskedLM",
|
||||||
"TFLayoutLMForSequenceClassification",
|
|
||||||
"TFLayoutLMForQuestionAnswering",
|
"TFLayoutLMForQuestionAnswering",
|
||||||
|
"TFLayoutLMForSequenceClassification",
|
||||||
"TFLayoutLMForTokenClassification",
|
"TFLayoutLMForTokenClassification",
|
||||||
"TFLayoutLMMainLayer",
|
"TFLayoutLMMainLayer",
|
||||||
"TFLayoutLMModel",
|
"TFLayoutLMModel",
|
||||||
@@ -2743,10 +2742,10 @@ else:
|
|||||||
_import_structure["models.mobilevit"].extend(
|
_import_structure["models.mobilevit"].extend(
|
||||||
[
|
[
|
||||||
"TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
"TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LIST",
|
||||||
"TFMobileViTPreTrainedModel",
|
|
||||||
"TFMobileViTModel",
|
|
||||||
"TFMobileViTForImageClassification",
|
"TFMobileViTForImageClassification",
|
||||||
"TFMobileViTForSemanticSegmentation",
|
"TFMobileViTForSemanticSegmentation",
|
||||||
|
"TFMobileViTModel",
|
||||||
|
"TFMobileViTPreTrainedModel",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
_import_structure["models.mpnet"].extend(
|
_import_structure["models.mpnet"].extend(
|
||||||
@@ -2999,11 +2998,11 @@ except OptionalDependencyNotAvailable:
|
|||||||
name for name in dir(dummy_flax_objects) if not name.startswith("_")
|
name for name in dir(dummy_flax_objects) if not name.startswith("_")
|
||||||
]
|
]
|
||||||
else:
|
else:
|
||||||
_import_structure["generation_flax_utils"] = []
|
|
||||||
_import_structure["generation"].extend(
|
_import_structure["generation"].extend(
|
||||||
[
|
[
|
||||||
"FlaxForcedBOSTokenLogitsProcessor",
|
"FlaxForcedBOSTokenLogitsProcessor",
|
||||||
"FlaxForcedEOSTokenLogitsProcessor",
|
"FlaxForcedEOSTokenLogitsProcessor",
|
||||||
|
"FlaxGenerationMixin",
|
||||||
"FlaxLogitsProcessor",
|
"FlaxLogitsProcessor",
|
||||||
"FlaxLogitsProcessorList",
|
"FlaxLogitsProcessorList",
|
||||||
"FlaxLogitsWarper",
|
"FlaxLogitsWarper",
|
||||||
@@ -3011,9 +3010,9 @@ else:
|
|||||||
"FlaxTemperatureLogitsWarper",
|
"FlaxTemperatureLogitsWarper",
|
||||||
"FlaxTopKLogitsWarper",
|
"FlaxTopKLogitsWarper",
|
||||||
"FlaxTopPLogitsWarper",
|
"FlaxTopPLogitsWarper",
|
||||||
"FlaxGenerationMixin",
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
_import_structure["generation_flax_utils"] = []
|
||||||
_import_structure["modeling_flax_outputs"] = []
|
_import_structure["modeling_flax_outputs"] = []
|
||||||
_import_structure["modeling_flax_utils"] = ["FlaxPreTrainedModel"]
|
_import_structure["modeling_flax_utils"] = ["FlaxPreTrainedModel"]
|
||||||
_import_structure["models.albert"].extend(
|
_import_structure["models.albert"].extend(
|
||||||
|
|||||||
@@ -47,8 +47,13 @@ except OptionalDependencyNotAvailable:
|
|||||||
else:
|
else:
|
||||||
_import_structure["feature_extraction_speech_to_text"] = ["Speech2TextFeatureExtractor"]
|
_import_structure["feature_extraction_speech_to_text"] = ["Speech2TextFeatureExtractor"]
|
||||||
|
|
||||||
if is_sentencepiece_available():
|
try:
|
||||||
_import_structure["processing_speech_to_text"] = ["Speech2TextProcessor"]
|
if not (is_speech_available() and is_sentencepiece_available()):
|
||||||
|
raise OptionalDependencyNotAvailable()
|
||||||
|
except OptionalDependencyNotAvailable:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
_import_structure["processing_speech_to_text"] = ["Speech2TextProcessor"]
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if not is_tf_available():
|
if not is_tf_available():
|
||||||
@@ -96,8 +101,13 @@ if TYPE_CHECKING:
|
|||||||
else:
|
else:
|
||||||
from .feature_extraction_speech_to_text import Speech2TextFeatureExtractor
|
from .feature_extraction_speech_to_text import Speech2TextFeatureExtractor
|
||||||
|
|
||||||
if is_sentencepiece_available():
|
try:
|
||||||
from .processing_speech_to_text import Speech2TextProcessor
|
if not (is_speech_available() and is_sentencepiece_available()):
|
||||||
|
raise OptionalDependencyNotAvailable()
|
||||||
|
except OptionalDependencyNotAvailable:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
from .processing_speech_to_text import Speech2TextProcessor
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if not is_tf_available():
|
if not is_tf_available():
|
||||||
|
|||||||
@@ -200,9 +200,9 @@ def sort_imports(file, check_only=True):
|
|||||||
indent = get_indent(block_lines[1])
|
indent = get_indent(block_lines[1])
|
||||||
# Slit the internal block into blocks of indent level 1.
|
# Slit the internal block into blocks of indent level 1.
|
||||||
internal_blocks = split_code_in_indented_blocks(internal_block_code, indent_level=indent)
|
internal_blocks = split_code_in_indented_blocks(internal_block_code, indent_level=indent)
|
||||||
# We have two categories of import key: list or _import_structu[key].append/extend
|
# We have two categories of import key: list or _import_structure[key].append/extend
|
||||||
pattern = _re_direct_key if "_import_structure" in block_lines[0] else _re_indirect_key
|
pattern = _re_direct_key if "_import_structure = {" in block_lines[0] else _re_indirect_key
|
||||||
# Grab the keys, but there is a trap: some lines are empty or jsut comments.
|
# Grab the keys, but there is a trap: some lines are empty or just comments.
|
||||||
keys = [(pattern.search(b).groups()[0] if pattern.search(b) is not None else None) for b in internal_blocks]
|
keys = [(pattern.search(b).groups()[0] if pattern.search(b) is not None else None) for b in internal_blocks]
|
||||||
# We only sort the lines with a key.
|
# We only sort the lines with a key.
|
||||||
keys_to_sort = [(i, key) for i, key in enumerate(keys) if key is not None]
|
keys_to_sort = [(i, key) for i, key in enumerate(keys) if key is not None]
|
||||||
|
|||||||
Reference in New Issue
Block a user