|
198 | 198 | "models.electra": ["ELECTRA_PRETRAINED_CONFIG_ARCHIVE_MAP", "ElectraConfig", "ElectraTokenizer"],
|
199 | 199 | "models.encoder_decoder": ["EncoderDecoderConfig"],
|
200 | 200 | "models.flaubert": ["FLAUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP", "FlaubertConfig", "FlaubertTokenizer"],
|
| 201 | + "models.flava": [ |
| 202 | + "FLAVA_PRETRAINED_CONFIG_ARCHIVE_MAP", |
| 203 | + "FlavaConfig", |
| 204 | + "FlavaImageCodebookConfig", |
| 205 | + "FlavaImageConfig", |
| 206 | + "FlavaMultimodalConfig", |
| 207 | + "FlavaTextConfig", |
| 208 | + ], |
201 | 209 | "models.fnet": ["FNET_PRETRAINED_CONFIG_ARCHIVE_MAP", "FNetConfig"],
|
202 | 210 | "models.fsmt": ["FSMT_PRETRAINED_CONFIG_ARCHIVE_MAP", "FSMTConfig", "FSMTTokenizer"],
|
203 | 211 | "models.funnel": ["FUNNEL_PRETRAINED_CONFIG_ARCHIVE_MAP", "FunnelConfig", "FunnelTokenizer"],
|
|
568 | 576 | _import_structure["models.deit"].append("DeiTFeatureExtractor")
|
569 | 577 | _import_structure["models.detr"].append("DetrFeatureExtractor")
|
570 | 578 | _import_structure["models.dpt"].append("DPTFeatureExtractor")
|
| 579 | + _import_structure["models.flava"].extend(["FlavaFeatureExtractor", "FlavaProcessor"]) |
571 | 580 | _import_structure["models.glpn"].append("GLPNFeatureExtractor")
|
572 | 581 | _import_structure["models.imagegpt"].append("ImageGPTFeatureExtractor")
|
573 | 582 | _import_structure["models.layoutlmv2"].append("LayoutLMv2FeatureExtractor")
|
|
1038 | 1047 | "FlaubertWithLMHeadModel",
|
1039 | 1048 | ]
|
1040 | 1049 | )
|
| 1050 | + _import_structure["models.flava"].extend( |
| 1051 | + [ |
| 1052 | + "FLAVA_PRETRAINED_MODEL_ARCHIVE_LIST", |
| 1053 | + "FlavaForPreTraining", |
| 1054 | + "FlavaImageCodebook", |
| 1055 | + "FlavaImageModel", |
| 1056 | + "FlavaModel", |
| 1057 | + "FlavaMultimodalModel", |
| 1058 | + "FlavaPreTrainedModel", |
| 1059 | + "FlavaTextModel", |
| 1060 | + ] |
| 1061 | + ) |
1041 | 1062 | _import_structure["models.fnet"].extend(
|
1042 | 1063 | [
|
1043 | 1064 | "FNET_PRETRAINED_MODEL_ARCHIVE_LIST",
|
|
2654 | 2675 | from .models.electra import ELECTRA_PRETRAINED_CONFIG_ARCHIVE_MAP, ElectraConfig, ElectraTokenizer
|
2655 | 2676 | from .models.encoder_decoder import EncoderDecoderConfig
|
2656 | 2677 | from .models.flaubert import FLAUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, FlaubertConfig, FlaubertTokenizer
|
| 2678 | + from .models.flava import ( |
| 2679 | + FLAVA_PRETRAINED_CONFIG_ARCHIVE_MAP, |
| 2680 | + FlavaConfig, |
| 2681 | + FlavaImageCodebookConfig, |
| 2682 | + FlavaImageConfig, |
| 2683 | + FlavaMultimodalConfig, |
| 2684 | + FlavaTextConfig, |
| 2685 | + ) |
2657 | 2686 | from .models.fnet import FNET_PRETRAINED_CONFIG_ARCHIVE_MAP, FNetConfig
|
2658 | 2687 | from .models.fsmt import FSMT_PRETRAINED_CONFIG_ARCHIVE_MAP, FSMTConfig, FSMTTokenizer
|
2659 | 2688 | from .models.funnel import FUNNEL_PRETRAINED_CONFIG_ARCHIVE_MAP, FunnelConfig, FunnelTokenizer
|
|
2974 | 3003 | from .models.deit import DeiTFeatureExtractor
|
2975 | 3004 | from .models.detr import DetrFeatureExtractor
|
2976 | 3005 | from .models.dpt import DPTFeatureExtractor
|
| 3006 | + from .models.flava import FlavaFeatureExtractor, FlavaProcessor |
2977 | 3007 | from .models.glpn import GLPNFeatureExtractor
|
2978 | 3008 | from .models.imagegpt import ImageGPTFeatureExtractor
|
2979 | 3009 | from .models.layoutlmv2 import LayoutLMv2FeatureExtractor, LayoutLMv2Processor
|
|
3372 | 3402 | FlaubertModel,
|
3373 | 3403 | FlaubertWithLMHeadModel,
|
3374 | 3404 | )
|
| 3405 | + from .models.flava import ( |
| 3406 | + FLAVA_PRETRAINED_MODEL_ARCHIVE_LIST, |
| 3407 | + FlavaForPreTraining, |
| 3408 | + FlavaImageCodebook, |
| 3409 | + FlavaImageModel, |
| 3410 | + FlavaModel, |
| 3411 | + FlavaMultimodalModel, |
| 3412 | + FlavaPreTrainedModel, |
| 3413 | + FlavaTextModel, |
| 3414 | + ) |
3375 | 3415 | from .models.fnet import (
|
3376 | 3416 | FNET_PRETRAINED_MODEL_ARCHIVE_LIST,
|
3377 | 3417 | FNetForMaskedLM,
|
|
0 commit comments