hf-transformers-bot commited on
Commit
1c583e4
·
verified ·
1 Parent(s): 5c773be

Update tiny models for PaddleOCRVLForConditionalGeneration

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
chat_template.jinja ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if not add_generation_prompt is defined -%}
2
+ {%- set add_generation_prompt = true -%}
3
+ {%- endif -%}
4
+ {%- if not cls_token is defined -%}
5
+ {%- set cls_token = "<|begin_of_sentence|>" -%}
6
+ {%- endif -%}
7
+ {%- if not eos_token is defined -%}
8
+ {%- set eos_token = "</s>" -%}
9
+ {%- endif -%}
10
+ {{- cls_token -}}
11
+ {%- for message in messages -%}
12
+ {%- if message["role"] == "user" -%}
13
+ {{- "User: " -}}
14
+ {%- for content in message["content"] -%}
15
+ {%- if content["type"] == "image" -%}
16
+ {{ "<|IMAGE_START|><|IMAGE_PLACEHOLDER|><|IMAGE_END|>" }}
17
+ {%- endif -%}
18
+ {%- endfor -%}
19
+ {%- for content in message["content"] -%}
20
+ {%- if content["type"] == "text" -%}
21
+ {{ content["text"] }}
22
+ {%- endif -%}
23
+ {%- endfor -%}
24
+ {{ "\n" -}}
25
+ {%- elif message["role"] == "assistant" -%}
26
+ {{- "Assistant: " -}}
27
+ {%- for content in message["content"] -%}
28
+ {%- if content["type"] == "text" -%}
29
+ {{ content["text"] }}
30
+ {%- endif -%}
31
+ {%- endfor -%}
32
+ {{ eos_token -}}
33
+ {%- elif message["role"] == "system" -%}
34
+ {%- for content in message["content"] -%}
35
+ {%- if content["type"] == "text" -%}
36
+ {{ content["text"] + "\n" }}
37
+ {%- endif -%}
38
+ {%- endfor -%}
39
+ {%- endif -%}
40
+ {%- endfor -%}
41
+ {%- if add_generation_prompt -%}
42
+ {{- "Assistant: " -}}
43
+ {%- endif -%}
config.json ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "PaddleOCRVLForConditionalGeneration"
4
+ ],
5
+ "dtype": "float32",
6
+ "image_token_id": 100295,
7
+ "model_type": "paddleocr_vl",
8
+ "text_config": {
9
+ "bos_token_id": 1,
10
+ "eos_token_id": 2,
11
+ "head_dim": 128,
12
+ "hidden_act": "silu",
13
+ "hidden_dropout_prob": 0.0,
14
+ "hidden_size": 32,
15
+ "ignored_index": -100,
16
+ "image_token_id": 100295,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 32,
19
+ "max_position_embeddings": 512,
20
+ "model_type": "paddleocr_vl_text",
21
+ "num_attention_heads": 4,
22
+ "num_hidden_layers": 2,
23
+ "num_key_value_heads": 2,
24
+ "pad_token_id": 0,
25
+ "rms_norm_eps": 1e-05,
26
+ "rope_parameters": {
27
+ "mrope_section": [
28
+ 16,
29
+ 24,
30
+ 24
31
+ ],
32
+ "rope_theta": 500000,
33
+ "rope_type": "default",
34
+ "type": "default"
35
+ },
36
+ "tie_word_embeddings": false,
37
+ "use_bias": false,
38
+ "use_cache": true,
39
+ "vocab_size": 101314
40
+ },
41
+ "tie_word_embeddings": true,
42
+ "transformers_version": "5.16.0.dev0",
43
+ "video_token_id": 100296,
44
+ "vision_config": {
45
+ "attention_dropout": 0.0,
46
+ "hidden_act": "gelu_pytorch_tanh",
47
+ "hidden_size": 144,
48
+ "image_size": 384,
49
+ "intermediate_size": 32,
50
+ "layer_norm_eps": 1e-06,
51
+ "model_type": "paddleocr_vl_vision",
52
+ "num_attention_heads": 4,
53
+ "num_channels": 3,
54
+ "num_hidden_layers": 2,
55
+ "pad_token_id": 0,
56
+ "patch_size": 14,
57
+ "spatial_merge_size": 2
58
+ },
59
+ "vision_end_token_id": 101306,
60
+ "vision_start_token_id": 101305
61
+ }
generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "eos_token_id": 2,
5
+ "output_attentions": false,
6
+ "output_hidden_states": false,
7
+ "pad_token_id": 0,
8
+ "transformers_version": "5.16.0.dev0",
9
+ "use_cache": true
10
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa51a0628fb3de7602dd3940d5296404199b77ca4b551fb24c4a9725d45aa169
3
+ size 16306528
preprocessor_config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": {
3
+ "longest_edge": 384,
4
+ "shortest_edge": 384
5
+ },
6
+ "do_convert_rgb": true,
7
+ "do_normalize": true,
8
+ "do_rescale": true,
9
+ "do_resize": true,
10
+ "image_mean": [
11
+ 0.48145466,
12
+ 0.4578275,
13
+ 0.40821073
14
+ ],
15
+ "image_processor_type": "PaddleOCRVLImageProcessor",
16
+ "image_std": [
17
+ 0.26862954,
18
+ 0.26130258,
19
+ 0.27577711
20
+ ],
21
+ "merge_size": 2,
22
+ "patch_size": 14,
23
+ "resample": 3,
24
+ "rescale_factor": 0.00392156862745098,
25
+ "size": {
26
+ "longest_edge": 384,
27
+ "shortest_edge": 384
28
+ },
29
+ "temporal_patch_size": 1
30
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f90f04fd8e5eb6dfa380f37d10c87392de8438dccb6768a2486b5a96ee76dba6
3
+ size 11187679
tokenizer_config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "auto_map": {
4
+ "AutoProcessor": "processing_paddleocr_vl.PaddleOCRVLProcessor"
5
+ },
6
+ "backend": "tokenizers",
7
+ "bos_token": "<s>",
8
+ "clean_up_tokenization_spaces": false,
9
+ "cls_token": "<|begin_of_sentence|>",
10
+ "eos_token": "</s>",
11
+ "image_token": "<|IMAGE_PLACEHOLDER|>",
12
+ "is_local": true,
13
+ "legacy": true,
14
+ "local_files_only": false,
15
+ "mask_token": "<mask:1>",
16
+ "model_max_length": 131072,
17
+ "model_specific_special_tokens": {
18
+ "image_token": "<|IMAGE_PLACEHOLDER|>"
19
+ },
20
+ "pad_token": "<unk>",
21
+ "processor_class": "PaddleOCRVLProcessor",
22
+ "sep_token": "<|end_of_sentence|>",
23
+ "sp_model_kwargs": {},
24
+ "spaces_between_special_tokens": false,
25
+ "tokenizer_class": "TokenizersBackend",
26
+ "unk_token": "<unk>",
27
+ "use_default_system_prompt": false
28
+ }