hf-transformers-bot commited on
Commit
13e5114
·
verified ·
1 Parent(s): 12ed7a4

Update tiny models for ColPaliForRetrieval

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
config.json ADDED
@@ -0,0 +1,161 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ColPaliForRetrieval"
4
+ ],
5
+ "dtype": "float32",
6
+ "embedding_dim": 128,
7
+ "model_type": "colpali",
8
+ "text_config": {
9
+ "attention_bias": false,
10
+ "attention_dropout": 0.0,
11
+ "attention_probs_dropout_prob": 0.1,
12
+ "bos_token_id": 2,
13
+ "eos_token_id": 1,
14
+ "head_dim": 8,
15
+ "hidden_act": "gelu_pytorch_tanh",
16
+ "hidden_activation": "gelu_pytorch_tanh",
17
+ "hidden_dropout_prob": 0.1,
18
+ "hidden_size": 32,
19
+ "id2label": {
20
+ "0": "LABEL_0",
21
+ "1": "LABEL_1",
22
+ "2": "LABEL_2"
23
+ },
24
+ "initializer_range": 0.02,
25
+ "intermediate_size": 37,
26
+ "is_training": true,
27
+ "label2id": {
28
+ "LABEL_0": 0,
29
+ "LABEL_1": 1,
30
+ "LABEL_2": 2
31
+ },
32
+ "max_position_embeddings": 512,
33
+ "model_type": "gemma",
34
+ "num_attention_heads": 4,
35
+ "num_choices": 4,
36
+ "num_hidden_layers": 2,
37
+ "num_image_tokens": 16,
38
+ "num_key_value_heads": 1,
39
+ "pad_token_id": 0,
40
+ "rms_norm_eps": 1e-06,
41
+ "rope_parameters": {
42
+ "rope_theta": 10000.0,
43
+ "rope_type": "default"
44
+ },
45
+ "seq_length": 128,
46
+ "tie_word_embeddings": true,
47
+ "type_sequence_label_size": 2,
48
+ "type_vocab_size": 16,
49
+ "use_bidirectional_attention": true,
50
+ "use_cache": true,
51
+ "use_labels": true,
52
+ "use_token_type_ids": false,
53
+ "vocab_size": 257153
54
+ },
55
+ "transformers_version": "5.16.0.dev0",
56
+ "vlm_config": {
57
+ "hidden_size": 2048,
58
+ "ignore_index": -100,
59
+ "image_token_index": 0,
60
+ "model_type": "paligemma",
61
+ "projection_dim": 32,
62
+ "projector_hidden_act": "gelu",
63
+ "text_config": {
64
+ "_name_or_path": "",
65
+ "architectures": null,
66
+ "attention_bias": false,
67
+ "attention_dropout": 0.0,
68
+ "attention_probs_dropout_prob": 0.1,
69
+ "bos_token_id": 2,
70
+ "chunk_size_feed_forward": 0,
71
+ "dtype": null,
72
+ "eos_token_id": 1,
73
+ "head_dim": 8,
74
+ "hidden_act": "gelu_pytorch_tanh",
75
+ "hidden_activation": "gelu_pytorch_tanh",
76
+ "hidden_dropout_prob": 0.1,
77
+ "hidden_size": 32,
78
+ "id2label": {
79
+ "0": "LABEL_0",
80
+ "1": "LABEL_1",
81
+ "2": "LABEL_2"
82
+ },
83
+ "initializer_range": 0.02,
84
+ "intermediate_size": 37,
85
+ "is_encoder_decoder": false,
86
+ "is_training": true,
87
+ "label2id": {
88
+ "LABEL_0": 0,
89
+ "LABEL_1": 1,
90
+ "LABEL_2": 2
91
+ },
92
+ "max_position_embeddings": 512,
93
+ "model_type": "gemma",
94
+ "num_attention_heads": 4,
95
+ "num_choices": 4,
96
+ "num_hidden_layers": 2,
97
+ "num_image_tokens": 16,
98
+ "num_key_value_heads": 1,
99
+ "output_attentions": false,
100
+ "output_hidden_states": false,
101
+ "pad_token_id": 0,
102
+ "problem_type": null,
103
+ "return_dict": true,
104
+ "rms_norm_eps": 1e-06,
105
+ "rope_parameters": {
106
+ "rope_theta": 10000.0,
107
+ "rope_type": "default"
108
+ },
109
+ "seq_length": 128,
110
+ "tie_word_embeddings": true,
111
+ "type_sequence_label_size": 2,
112
+ "type_vocab_size": 16,
113
+ "use_bidirectional_attention": true,
114
+ "use_cache": true,
115
+ "use_labels": true,
116
+ "use_token_type_ids": false,
117
+ "vocab_size": 257153
118
+ },
119
+ "tie_word_embeddings": true,
120
+ "vision_config": {
121
+ "_name_or_path": "",
122
+ "architectures": null,
123
+ "attention_dropout": 0.1,
124
+ "chunk_size_feed_forward": 0,
125
+ "dropout": 0.1,
126
+ "dtype": null,
127
+ "hidden_act": "gelu_pytorch_tanh",
128
+ "hidden_size": 32,
129
+ "id2label": {
130
+ "0": "LABEL_0",
131
+ "1": "LABEL_1"
132
+ },
133
+ "image_size": 20,
134
+ "initializer_range": 0.02,
135
+ "intermediate_size": 37,
136
+ "is_encoder_decoder": false,
137
+ "is_training": true,
138
+ "label2id": {
139
+ "LABEL_0": 0,
140
+ "LABEL_1": 1
141
+ },
142
+ "layer_norm_eps": 1e-06,
143
+ "model_type": "siglip_vision_model",
144
+ "num_attention_heads": 4,
145
+ "num_channels": 3,
146
+ "num_hidden_layers": 2,
147
+ "num_image_tokens": 4,
148
+ "num_key_value_heads": 1,
149
+ "output_attentions": false,
150
+ "output_hidden_states": false,
151
+ "patch_size": 5,
152
+ "problem_type": null,
153
+ "projection_dim": 32,
154
+ "return_dict": true,
155
+ "use_labels": true
156
+ },
157
+ "vision_feature_layer": -1,
158
+ "vision_feature_select_strategy": "default",
159
+ "vocab_size": 257152
160
+ }
161
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40d046545c418368bf5e86d5fe10036d4ed2f934d9e33094ea3b530ecc7fd334
3
+ size 33088132
preprocessor_config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_rgb": null,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.5,
8
+ 0.5,
9
+ 0.5
10
+ ],
11
+ "image_processor_type": "SiglipImageProcessor",
12
+ "image_seq_length": 1024,
13
+ "image_std": [
14
+ 0.5,
15
+ 0.5,
16
+ 0.5
17
+ ],
18
+ "resample": 3,
19
+ "rescale_factor": 0.00392156862745098,
20
+ "size": {
21
+ "height": 448,
22
+ "width": 448
23
+ }
24
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a0719264e920323136a27a14fbed84870f400c3d91dab7e813deba64d639fb8f
3
+ size 34600974
tokenizer_config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "bos_token": "<bos>",
4
+ "clean_up_tokenization_spaces": false,
5
+ "eos_token": "<eos>",
6
+ "extra_special_tokens": [
7
+ "<image>"
8
+ ],
9
+ "is_local": true,
10
+ "local_files_only": false,
11
+ "mask_token": "<mask>",
12
+ "max_length": 50,
13
+ "model_max_length": 1000000000000000019884624838656,
14
+ "pad_to_multiple_of": null,
15
+ "pad_token": "<pad>",
16
+ "pad_token_type_id": 0,
17
+ "padding_side": "right",
18
+ "processor_class": "PaliGemmaProcessor",
19
+ "sp_model_kwargs": {},
20
+ "spaces_between_special_tokens": false,
21
+ "tokenizer_class": "GemmaTokenizer",
22
+ "unk_token": "<unk>",
23
+ "use_default_system_prompt": false
24
+ }