Spaces:
Running
on
T4
Running
on
T4
| # Copyright 2020 The HuggingFace Team. All rights reserved. | |
| # | |
| # Licensed under the Apache License, Version 2.0 (the "License"); | |
| # you may not use this file except in compliance with the License. | |
| # You may obtain a copy of the License at | |
| # | |
| # http://www.apache.org/licenses/LICENSE-2.0 | |
| # | |
| # Unless required by applicable law or agreed to in writing, software | |
| # distributed under the License is distributed on an "AS IS" BASIS, | |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| # See the License for the specific language governing permissions and | |
| # limitations under the License. | |
| from typing import TYPE_CHECKING | |
| from ...utils import ( | |
| OptionalDependencyNotAvailable, | |
| _LazyModule, | |
| is_flax_available, | |
| is_tf_available, | |
| is_torch_available, | |
| ) | |
| _import_structure = { | |
| "auto_factory": ["get_values"], | |
| "configuration_auto": ["ALL_PRETRAINED_CONFIG_ARCHIVE_MAP", "CONFIG_MAPPING", "MODEL_NAMES_MAPPING", "AutoConfig"], | |
| "feature_extraction_auto": ["FEATURE_EXTRACTOR_MAPPING", "AutoFeatureExtractor"], | |
| "image_processing_auto": ["IMAGE_PROCESSOR_MAPPING", "AutoImageProcessor"], | |
| "processing_auto": ["PROCESSOR_MAPPING", "AutoProcessor"], | |
| "tokenization_auto": ["TOKENIZER_MAPPING", "AutoTokenizer"], | |
| } | |
| try: | |
| if not is_torch_available(): | |
| raise OptionalDependencyNotAvailable() | |
| except OptionalDependencyNotAvailable: | |
| pass | |
| else: | |
| _import_structure["modeling_auto"] = [ | |
| "MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING", | |
| "MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPING", | |
| "MODEL_FOR_AUDIO_XVECTOR_MAPPING", | |
| "MODEL_FOR_BACKBONE_MAPPING", | |
| "MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPING", | |
| "MODEL_FOR_CAUSAL_LM_MAPPING", | |
| "MODEL_FOR_CTC_MAPPING", | |
| "MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING", | |
| "MODEL_FOR_DEPTH_ESTIMATION_MAPPING", | |
| "MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING", | |
| "MODEL_FOR_IMAGE_SEGMENTATION_MAPPING", | |
| "MODEL_FOR_IMAGE_TO_IMAGE_MAPPING", | |
| "MODEL_FOR_INSTANCE_SEGMENTATION_MAPPING", | |
| "MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING", | |
| "MODEL_FOR_MASKED_LM_MAPPING", | |
| "MODEL_FOR_MASK_GENERATION_MAPPING", | |
| "MODEL_FOR_MULTIPLE_CHOICE_MAPPING", | |
| "MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING", | |
| "MODEL_FOR_OBJECT_DETECTION_MAPPING", | |
| "MODEL_FOR_PRETRAINING_MAPPING", | |
| "MODEL_FOR_QUESTION_ANSWERING_MAPPING", | |
| "MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING", | |
| "MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING", | |
| "MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING", | |
| "MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING", | |
| "MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING", | |
| "MODEL_FOR_TEXT_ENCODING_MAPPING", | |
| "MODEL_FOR_TEXT_TO_WAVEFORM_MAPPING", | |
| "MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPING", | |
| "MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING", | |
| "MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPING", | |
| "MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING", | |
| "MODEL_FOR_VISION_2_SEQ_MAPPING", | |
| "MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPING", | |
| "MODEL_MAPPING", | |
| "MODEL_WITH_LM_HEAD_MAPPING", | |
| "MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPING", | |
| "MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPING", | |
| "AutoModel", | |
| "AutoBackbone", | |
| "AutoModelForAudioClassification", | |
| "AutoModelForAudioFrameClassification", | |
| "AutoModelForAudioXVector", | |
| "AutoModelForCausalLM", | |
| "AutoModelForCTC", | |
| "AutoModelForDepthEstimation", | |
| "AutoModelForImageClassification", | |
| "AutoModelForImageSegmentation", | |
| "AutoModelForImageToImage", | |
| "AutoModelForInstanceSegmentation", | |
| "AutoModelForMaskGeneration", | |
| "AutoModelForTextEncoding", | |
| "AutoModelForMaskedImageModeling", | |
| "AutoModelForMaskedLM", | |
| "AutoModelForMultipleChoice", | |
| "AutoModelForNextSentencePrediction", | |
| "AutoModelForObjectDetection", | |
| "AutoModelForPreTraining", | |
| "AutoModelForQuestionAnswering", | |
| "AutoModelForSemanticSegmentation", | |
| "AutoModelForSeq2SeqLM", | |
| "AutoModelForSequenceClassification", | |
| "AutoModelForSpeechSeq2Seq", | |
| "AutoModelForTableQuestionAnswering", | |
| "AutoModelForTextToSpectrogram", | |
| "AutoModelForTextToWaveform", | |
| "AutoModelForTokenClassification", | |
| "AutoModelForUniversalSegmentation", | |
| "AutoModelForVideoClassification", | |
| "AutoModelForVision2Seq", | |
| "AutoModelForVisualQuestionAnswering", | |
| "AutoModelForDocumentQuestionAnswering", | |
| "AutoModelWithLMHead", | |
| "AutoModelForZeroShotImageClassification", | |
| "AutoModelForZeroShotObjectDetection", | |
| ] | |
| try: | |
| if not is_tf_available(): | |
| raise OptionalDependencyNotAvailable() | |
| except OptionalDependencyNotAvailable: | |
| pass | |
| else: | |
| _import_structure["modeling_tf_auto"] = [ | |
| "TF_MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING", | |
| "TF_MODEL_FOR_CAUSAL_LM_MAPPING", | |
| "TF_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING", | |
| "TF_MODEL_FOR_MASK_GENERATION_MAPPING", | |
| "TF_MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING", | |
| "TF_MODEL_FOR_MASKED_LM_MAPPING", | |
| "TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING", | |
| "TF_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING", | |
| "TF_MODEL_FOR_PRETRAINING_MAPPING", | |
| "TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING", | |
| "TF_MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING", | |
| "TF_MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING", | |
| "TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING", | |
| "TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING", | |
| "TF_MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING", | |
| "TF_MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING", | |
| "TF_MODEL_FOR_TEXT_ENCODING_MAPPING", | |
| "TF_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING", | |
| "TF_MODEL_FOR_VISION_2_SEQ_MAPPING", | |
| "TF_MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPING", | |
| "TF_MODEL_MAPPING", | |
| "TF_MODEL_WITH_LM_HEAD_MAPPING", | |
| "TFAutoModel", | |
| "TFAutoModelForAudioClassification", | |
| "TFAutoModelForCausalLM", | |
| "TFAutoModelForImageClassification", | |
| "TFAutoModelForMaskedImageModeling", | |
| "TFAutoModelForMaskedLM", | |
| "TFAutoModelForMaskGeneration", | |
| "TFAutoModelForMultipleChoice", | |
| "TFAutoModelForNextSentencePrediction", | |
| "TFAutoModelForPreTraining", | |
| "TFAutoModelForDocumentQuestionAnswering", | |
| "TFAutoModelForQuestionAnswering", | |
| "TFAutoModelForSemanticSegmentation", | |
| "TFAutoModelForSeq2SeqLM", | |
| "TFAutoModelForSequenceClassification", | |
| "TFAutoModelForSpeechSeq2Seq", | |
| "TFAutoModelForTableQuestionAnswering", | |
| "TFAutoModelForTextEncoding", | |
| "TFAutoModelForTokenClassification", | |
| "TFAutoModelForVision2Seq", | |
| "TFAutoModelForZeroShotImageClassification", | |
| "TFAutoModelWithLMHead", | |
| ] | |
| try: | |
| if not is_flax_available(): | |
| raise OptionalDependencyNotAvailable() | |
| except OptionalDependencyNotAvailable: | |
| pass | |
| else: | |
| _import_structure["modeling_flax_auto"] = [ | |
| "FLAX_MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING", | |
| "FLAX_MODEL_FOR_CAUSAL_LM_MAPPING", | |
| "FLAX_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING", | |
| "FLAX_MODEL_FOR_MASKED_LM_MAPPING", | |
| "FLAX_MODEL_FOR_MULTIPLE_CHOICE_MAPPING", | |
| "FLAX_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING", | |
| "FLAX_MODEL_FOR_PRETRAINING_MAPPING", | |
| "FLAX_MODEL_FOR_QUESTION_ANSWERING_MAPPING", | |
| "FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING", | |
| "FLAX_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING", | |
| "FLAX_MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING", | |
| "FLAX_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING", | |
| "FLAX_MODEL_FOR_VISION_2_SEQ_MAPPING", | |
| "FLAX_MODEL_MAPPING", | |
| "FlaxAutoModel", | |
| "FlaxAutoModelForCausalLM", | |
| "FlaxAutoModelForImageClassification", | |
| "FlaxAutoModelForMaskedLM", | |
| "FlaxAutoModelForMultipleChoice", | |
| "FlaxAutoModelForNextSentencePrediction", | |
| "FlaxAutoModelForPreTraining", | |
| "FlaxAutoModelForQuestionAnswering", | |
| "FlaxAutoModelForSeq2SeqLM", | |
| "FlaxAutoModelForSequenceClassification", | |
| "FlaxAutoModelForSpeechSeq2Seq", | |
| "FlaxAutoModelForTokenClassification", | |
| "FlaxAutoModelForVision2Seq", | |
| ] | |
| if TYPE_CHECKING: | |
| from .auto_factory import get_values | |
| from .configuration_auto import ALL_PRETRAINED_CONFIG_ARCHIVE_MAP, CONFIG_MAPPING, MODEL_NAMES_MAPPING, AutoConfig | |
| from .feature_extraction_auto import FEATURE_EXTRACTOR_MAPPING, AutoFeatureExtractor | |
| from .image_processing_auto import IMAGE_PROCESSOR_MAPPING, AutoImageProcessor | |
| from .processing_auto import PROCESSOR_MAPPING, AutoProcessor | |
| from .tokenization_auto import TOKENIZER_MAPPING, AutoTokenizer | |
| try: | |
| if not is_torch_available(): | |
| raise OptionalDependencyNotAvailable() | |
| except OptionalDependencyNotAvailable: | |
| pass | |
| else: | |
| from .modeling_auto import ( | |
| MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING, | |
| MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPING, | |
| MODEL_FOR_AUDIO_XVECTOR_MAPPING, | |
| MODEL_FOR_BACKBONE_MAPPING, | |
| MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPING, | |
| MODEL_FOR_CAUSAL_LM_MAPPING, | |
| MODEL_FOR_CTC_MAPPING, | |
| MODEL_FOR_DEPTH_ESTIMATION_MAPPING, | |
| MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING, | |
| MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING, | |
| MODEL_FOR_IMAGE_SEGMENTATION_MAPPING, | |
| MODEL_FOR_IMAGE_TO_IMAGE_MAPPING, | |
| MODEL_FOR_INSTANCE_SEGMENTATION_MAPPING, | |
| MODEL_FOR_MASK_GENERATION_MAPPING, | |
| MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING, | |
| MODEL_FOR_MASKED_LM_MAPPING, | |
| MODEL_FOR_MULTIPLE_CHOICE_MAPPING, | |
| MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING, | |
| MODEL_FOR_OBJECT_DETECTION_MAPPING, | |
| MODEL_FOR_PRETRAINING_MAPPING, | |
| MODEL_FOR_QUESTION_ANSWERING_MAPPING, | |
| MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING, | |
| MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING, | |
| MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING, | |
| MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING, | |
| MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING, | |
| MODEL_FOR_TEXT_ENCODING_MAPPING, | |
| MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPING, | |
| MODEL_FOR_TEXT_TO_WAVEFORM_MAPPING, | |
| MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING, | |
| MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPING, | |
| MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING, | |
| MODEL_FOR_VISION_2_SEQ_MAPPING, | |
| MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPING, | |
| MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPING, | |
| MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPING, | |
| MODEL_MAPPING, | |
| MODEL_WITH_LM_HEAD_MAPPING, | |
| AutoBackbone, | |
| AutoModel, | |
| AutoModelForAudioClassification, | |
| AutoModelForAudioFrameClassification, | |
| AutoModelForAudioXVector, | |
| AutoModelForCausalLM, | |
| AutoModelForCTC, | |
| AutoModelForDepthEstimation, | |
| AutoModelForDocumentQuestionAnswering, | |
| AutoModelForImageClassification, | |
| AutoModelForImageSegmentation, | |
| AutoModelForImageToImage, | |
| AutoModelForInstanceSegmentation, | |
| AutoModelForMaskedImageModeling, | |
| AutoModelForMaskedLM, | |
| AutoModelForMaskGeneration, | |
| AutoModelForMultipleChoice, | |
| AutoModelForNextSentencePrediction, | |
| AutoModelForObjectDetection, | |
| AutoModelForPreTraining, | |
| AutoModelForQuestionAnswering, | |
| AutoModelForSemanticSegmentation, | |
| AutoModelForSeq2SeqLM, | |
| AutoModelForSequenceClassification, | |
| AutoModelForSpeechSeq2Seq, | |
| AutoModelForTableQuestionAnswering, | |
| AutoModelForTextEncoding, | |
| AutoModelForTextToSpectrogram, | |
| AutoModelForTextToWaveform, | |
| AutoModelForTokenClassification, | |
| AutoModelForUniversalSegmentation, | |
| AutoModelForVideoClassification, | |
| AutoModelForVision2Seq, | |
| AutoModelForVisualQuestionAnswering, | |
| AutoModelForZeroShotImageClassification, | |
| AutoModelForZeroShotObjectDetection, | |
| AutoModelWithLMHead, | |
| ) | |
| try: | |
| if not is_tf_available(): | |
| raise OptionalDependencyNotAvailable() | |
| except OptionalDependencyNotAvailable: | |
| pass | |
| else: | |
| from .modeling_tf_auto import ( | |
| TF_MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING, | |
| TF_MODEL_FOR_CAUSAL_LM_MAPPING, | |
| TF_MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING, | |
| TF_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING, | |
| TF_MODEL_FOR_MASK_GENERATION_MAPPING, | |
| TF_MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING, | |
| TF_MODEL_FOR_MASKED_LM_MAPPING, | |
| TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING, | |
| TF_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING, | |
| TF_MODEL_FOR_PRETRAINING_MAPPING, | |
| TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING, | |
| TF_MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING, | |
| TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING, | |
| TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING, | |
| TF_MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING, | |
| TF_MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING, | |
| TF_MODEL_FOR_TEXT_ENCODING_MAPPING, | |
| TF_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING, | |
| TF_MODEL_FOR_VISION_2_SEQ_MAPPING, | |
| TF_MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPING, | |
| TF_MODEL_MAPPING, | |
| TF_MODEL_WITH_LM_HEAD_MAPPING, | |
| TFAutoModel, | |
| TFAutoModelForAudioClassification, | |
| TFAutoModelForCausalLM, | |
| TFAutoModelForDocumentQuestionAnswering, | |
| TFAutoModelForImageClassification, | |
| TFAutoModelForMaskedImageModeling, | |
| TFAutoModelForMaskedLM, | |
| TFAutoModelForMaskGeneration, | |
| TFAutoModelForMultipleChoice, | |
| TFAutoModelForNextSentencePrediction, | |
| TFAutoModelForPreTraining, | |
| TFAutoModelForQuestionAnswering, | |
| TFAutoModelForSemanticSegmentation, | |
| TFAutoModelForSeq2SeqLM, | |
| TFAutoModelForSequenceClassification, | |
| TFAutoModelForSpeechSeq2Seq, | |
| TFAutoModelForTableQuestionAnswering, | |
| TFAutoModelForTextEncoding, | |
| TFAutoModelForTokenClassification, | |
| TFAutoModelForVision2Seq, | |
| TFAutoModelForZeroShotImageClassification, | |
| TFAutoModelWithLMHead, | |
| ) | |
| try: | |
| if not is_flax_available(): | |
| raise OptionalDependencyNotAvailable() | |
| except OptionalDependencyNotAvailable: | |
| pass | |
| else: | |
| from .modeling_flax_auto import ( | |
| FLAX_MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING, | |
| FLAX_MODEL_FOR_CAUSAL_LM_MAPPING, | |
| FLAX_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING, | |
| FLAX_MODEL_FOR_MASKED_LM_MAPPING, | |
| FLAX_MODEL_FOR_MULTIPLE_CHOICE_MAPPING, | |
| FLAX_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING, | |
| FLAX_MODEL_FOR_PRETRAINING_MAPPING, | |
| FLAX_MODEL_FOR_QUESTION_ANSWERING_MAPPING, | |
| FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING, | |
| FLAX_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING, | |
| FLAX_MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING, | |
| FLAX_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING, | |
| FLAX_MODEL_FOR_VISION_2_SEQ_MAPPING, | |
| FLAX_MODEL_MAPPING, | |
| FlaxAutoModel, | |
| FlaxAutoModelForCausalLM, | |
| FlaxAutoModelForImageClassification, | |
| FlaxAutoModelForMaskedLM, | |
| FlaxAutoModelForMultipleChoice, | |
| FlaxAutoModelForNextSentencePrediction, | |
| FlaxAutoModelForPreTraining, | |
| FlaxAutoModelForQuestionAnswering, | |
| FlaxAutoModelForSeq2SeqLM, | |
| FlaxAutoModelForSequenceClassification, | |
| FlaxAutoModelForSpeechSeq2Seq, | |
| FlaxAutoModelForTokenClassification, | |
| FlaxAutoModelForVision2Seq, | |
| ) | |
| else: | |
| import sys | |
| sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) | |