visfocus-base-docvqa / config.json
ofirab's picture
Upload model
3d1911b verified
Raw
History Blame Contribute Delete
2.16 kB
{
"architectures": [
"VisFocusModelForImageTextToText"
],
"auto_map": {
"AutoConfig": "configuration_visfocus.VisFocusConfig",
"AutoModel": "configuration_visfocus.VisFocusPreTrainedModel",
"AutoModelForConditionalGeneration": "configuration_visfocus.VisFocusModelForImageTextToText",
"AutoModelForImageTextToText": "modeling_visfocus.VisFocusModelForImageTextToText"
},
"cache_dir": null,
"do_lower_case": true,
"freeze_modules": [],
"generate_max_new_tokens_len": 256,
"hidden_dropout_prob": 0.1,
"image_size": [
1536,
768
],
"initializer_factor": "vf-base",
"initializer_range": 0.02,
"lm_config": {
"model_type": "t5"
},
"lora": null,
"matcher_type": "default",
"max_seq_length": 2048,
"model_name_or_path": "",
"model_type": "visfocus",
"seed": 42,
"torch_dtype": "float32",
"transformers_version": "4.46.0.dev0",
"unfreeze_modules": [],
"variant": "vf-base",
"vision": {
"model": {
"drop_path_rate": 0.3,
"drop_rate": 0.0,
"image_size": [
1536,
768
],
"name": "swin_small_patch4_window7_224_22k",
"swinv2": {
"ape": false,
"depths": [
2,
2,
18,
2
],
"downsampling_method": "merge_attention_v3",
"embed_dim": 96,
"in_chans": 3,
"lm_d_model": 768,
"mlp_ratio": 4.0,
"num_heads": [
3,
6,
12,
24
],
"patch_norm": true,
"patch_size": 4,
"pretrained_window_sizes": [
0,
0,
0,
0
],
"qkv_bias": true,
"text_embedder": "t5-base",
"vl_alpha": 0.5,
"vl_cross_attn_layers": [
3
],
"window_size": 24
},
"type": "swinv2",
"vision_resume_from": "https://github.com/SwinTransformer/storage/releases/download/v2.0.0/swinv2_small_patch4_window16_256.pth"
},
"train": {
"use_checkpoint": false
}
},
"vision_config": {
"model_type": "swin_vilma"
},
"vl_l1_loss": null,
"vqa_method": null
}