add epoch 1
Browse files- config.json +4 -0
- flax_model.msgpack +1 -1
config.json
CHANGED
|
@@ -2,6 +2,9 @@
|
|
| 2 |
"architectures": [
|
| 3 |
"ViTGPT2LMForConditionalGeneration"
|
| 4 |
],
|
|
|
|
|
|
|
|
|
|
| 5 |
"gpt2_config": {
|
| 6 |
"_name_or_path": ".",
|
| 7 |
"activation_function": "gelu_new",
|
|
@@ -90,6 +93,7 @@
|
|
| 90 |
},
|
| 91 |
"is_encoder_decoder": true,
|
| 92 |
"model_type": "vit-gpt2",
|
|
|
|
| 93 |
"transformers_version": null,
|
| 94 |
"vit_config": {
|
| 95 |
"_name_or_path": "",
|
|
|
|
| 2 |
"architectures": [
|
| 3 |
"ViTGPT2LMForConditionalGeneration"
|
| 4 |
],
|
| 5 |
+
"bos_token_id": 50256,
|
| 6 |
+
"decoder_start_token_id": 50256,
|
| 7 |
+
"eos_token_id": 50256,
|
| 8 |
"gpt2_config": {
|
| 9 |
"_name_or_path": ".",
|
| 10 |
"activation_function": "gelu_new",
|
|
|
|
| 93 |
},
|
| 94 |
"is_encoder_decoder": true,
|
| 95 |
"model_type": "vit-gpt2",
|
| 96 |
+
"pad_token_id": 50256,
|
| 97 |
"transformers_version": null,
|
| 98 |
"vit_config": {
|
| 99 |
"_name_or_path": "",
|
flax_model.msgpack
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1013496087
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6f94c730643e9a5fb08b2df00124daab4d2f8ef1e3a46359ef6d27961ed192ef
|
| 3 |
size 1013496087
|