Version 3
Browse filesNew finetuning using more examples
- config.json +4 -2
- generation_config.json +2 -1
- model.safetensors +1 -1
config.json
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
{
|
| 2 |
-
"_name_or_path": "
|
| 3 |
"architectures": [
|
| 4 |
"VisionEncoderDecoderModel"
|
| 5 |
],
|
| 6 |
"decoder": {
|
|
|
|
| 7 |
"_name_or_path": "",
|
| 8 |
"activation_dropout": 0.0,
|
| 9 |
"activation_function": "relu",
|
|
@@ -85,6 +86,7 @@
|
|
| 85 |
"vocab_size": 50265
|
| 86 |
},
|
| 87 |
"encoder": {
|
|
|
|
| 88 |
"_name_or_path": "",
|
| 89 |
"add_cross_attention": false,
|
| 90 |
"architectures": null,
|
|
@@ -165,5 +167,5 @@
|
|
| 165 |
"model_type": "vision-encoder-decoder",
|
| 166 |
"tie_word_embeddings": false,
|
| 167 |
"torch_dtype": "float32",
|
| 168 |
-
"transformers_version": "4.
|
| 169 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"_name_or_path": "../finetuned/trocr/Terminus/V_2_final",
|
| 3 |
"architectures": [
|
| 4 |
"VisionEncoderDecoderModel"
|
| 5 |
],
|
| 6 |
"decoder": {
|
| 7 |
+
"_attn_implementation_autoset": false,
|
| 8 |
"_name_or_path": "",
|
| 9 |
"activation_dropout": 0.0,
|
| 10 |
"activation_function": "relu",
|
|
|
|
| 86 |
"vocab_size": 50265
|
| 87 |
},
|
| 88 |
"encoder": {
|
| 89 |
+
"_attn_implementation_autoset": false,
|
| 90 |
"_name_or_path": "",
|
| 91 |
"add_cross_attention": false,
|
| 92 |
"architectures": null,
|
|
|
|
| 167 |
"model_type": "vision-encoder-decoder",
|
| 168 |
"tie_word_embeddings": false,
|
| 169 |
"torch_dtype": "float32",
|
| 170 |
+
"transformers_version": "4.46.3"
|
| 171 |
}
|
generation_config.json
CHANGED
|
@@ -4,8 +4,9 @@
|
|
| 4 |
"decoder_start_token_id": 0,
|
| 5 |
"do_sample": true,
|
| 6 |
"eos_token_id": 2,
|
|
|
|
| 7 |
"pad_token_id": 1,
|
| 8 |
"temperature": 0.4,
|
| 9 |
-
"transformers_version": "4.
|
| 10 |
"use_cache": false
|
| 11 |
}
|
|
|
|
| 4 |
"decoder_start_token_id": 0,
|
| 5 |
"do_sample": true,
|
| 6 |
"eos_token_id": 2,
|
| 7 |
+
"max_new_tokens": 120,
|
| 8 |
"pad_token_id": 1,
|
| 9 |
"temperature": 0.4,
|
| 10 |
+
"transformers_version": "4.46.3",
|
| 11 |
"use_cache": false
|
| 12 |
}
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 2436757100
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:699d5c6f54fa47acc1809e4c5c581f60208abd4fdf67a24e299dd16f5b7a36dd
|
| 3 |
size 2436757100
|