Upload TrOCR model for Ethiopic OCR
Browse files- config.json +7 -9
- generation_config.json +1 -1
- model.safetensors +1 -1
config.json
CHANGED
@@ -1,10 +1,9 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "/
|
3 |
"architectures": [
|
4 |
"VisionEncoderDecoderModel"
|
5 |
],
|
6 |
"decoder": {
|
7 |
-
"_attn_implementation_autoset": false,
|
8 |
"_name_or_path": "",
|
9 |
"activation_dropout": 0.0,
|
10 |
"activation_function": "gelu",
|
@@ -85,9 +84,8 @@
|
|
85 |
"vocab_size": 50265
|
86 |
},
|
87 |
"decoder_start_token_id": 0,
|
88 |
-
"early_stopping":
|
89 |
"encoder": {
|
90 |
-
"_attn_implementation_autoset": false,
|
91 |
"_name_or_path": "",
|
92 |
"add_cross_attention": false,
|
93 |
"architectures": null,
|
@@ -166,15 +164,15 @@
|
|
166 |
},
|
167 |
"eos_token_id": 2,
|
168 |
"is_encoder_decoder": true,
|
169 |
-
"length_penalty":
|
170 |
-
"max_length":
|
171 |
"model_type": "vision-encoder-decoder",
|
172 |
-
"no_repeat_ngram_size":
|
173 |
-
"num_beams":
|
174 |
"pad_token_id": 1,
|
175 |
"processor_class": "TrOCRProcessor",
|
176 |
"tie_word_embeddings": false,
|
177 |
"torch_dtype": "float32",
|
178 |
-
"transformers_version": "4.
|
179 |
"vocab_size": 50265
|
180 |
}
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "microsoft/trocr-base-handwritten",
|
3 |
"architectures": [
|
4 |
"VisionEncoderDecoderModel"
|
5 |
],
|
6 |
"decoder": {
|
|
|
7 |
"_name_or_path": "",
|
8 |
"activation_dropout": 0.0,
|
9 |
"activation_function": "gelu",
|
|
|
84 |
"vocab_size": 50265
|
85 |
},
|
86 |
"decoder_start_token_id": 0,
|
87 |
+
"early_stopping": true,
|
88 |
"encoder": {
|
|
|
89 |
"_name_or_path": "",
|
90 |
"add_cross_attention": false,
|
91 |
"architectures": null,
|
|
|
164 |
},
|
165 |
"eos_token_id": 2,
|
166 |
"is_encoder_decoder": true,
|
167 |
+
"length_penalty": 2.0,
|
168 |
+
"max_length": 64,
|
169 |
"model_type": "vision-encoder-decoder",
|
170 |
+
"no_repeat_ngram_size": 3,
|
171 |
+
"num_beams": 2,
|
172 |
"pad_token_id": 1,
|
173 |
"processor_class": "TrOCRProcessor",
|
174 |
"tie_word_embeddings": false,
|
175 |
"torch_dtype": "float32",
|
176 |
+
"transformers_version": "4.44.0",
|
177 |
"vocab_size": 50265
|
178 |
}
|
generation_config.json
CHANGED
@@ -8,6 +8,6 @@
|
|
8 |
"no_repeat_ngram_size": 3,
|
9 |
"num_beams": 2,
|
10 |
"pad_token_id": 1,
|
11 |
-
"transformers_version": "4.
|
12 |
"use_cache": false
|
13 |
}
|
|
|
8 |
"no_repeat_ngram_size": 3,
|
9 |
"num_beams": 2,
|
10 |
"pad_token_id": 1,
|
11 |
+
"transformers_version": "4.44.0",
|
12 |
"use_cache": false
|
13 |
}
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 1335747032
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2b5350315591cb29870e57500db30a2ebd067891319eadeab8959d6caedda5cc
|
3 |
size 1335747032
|