PhucMap commited on
Commit
b352616
·
1 Parent(s): 8136960

Upload MBartForConditionalGeneration

Browse files
Files changed (3) hide show
  1. config.json +10 -10
  2. generation_config.json +1 -1
  3. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/content/drive/MyDrive/BARTpho-finetuned-vi_to_vsl/checkpoint-800",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "architectures": [
@@ -8,17 +8,17 @@
8
  "attention_dropout": 0.0,
9
  "bos_token_id": 0,
10
  "classifier_dropout": 0.0,
11
- "d_model": 1024,
12
- "decoder_attention_heads": 16,
13
- "decoder_ffn_dim": 4096,
14
  "decoder_layerdrop": 0.0,
15
- "decoder_layers": 12,
16
  "decoder_start_token_id": 2,
17
  "dropout": 0.1,
18
- "encoder_attention_heads": 16,
19
- "encoder_ffn_dim": 4096,
20
  "encoder_layerdrop": 0.0,
21
- "encoder_layers": 12,
22
  "eos_token_id": 2,
23
  "forced_eos_token_id": 2,
24
  "gradient_checkpointing": false,
@@ -26,12 +26,12 @@
26
  "is_encoder_decoder": true,
27
  "max_position_embeddings": 1024,
28
  "model_type": "mbart",
29
- "num_hidden_layers": 12,
30
  "pad_token_id": 1,
31
  "scale_embedding": false,
32
  "tokenizer_class": "BartphoTokenizer",
33
  "torch_dtype": "float32",
34
- "transformers_version": "4.31.0",
35
  "use_cache": true,
36
  "vocab_size": 40030
37
  }
 
1
  {
2
+ "_name_or_path": "/content/drive/MyDrive/BARTpho-finetuned-vi_to_vsl/checkpoint-4400",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "architectures": [
 
8
  "attention_dropout": 0.0,
9
  "bos_token_id": 0,
10
  "classifier_dropout": 0.0,
11
+ "d_model": 768,
12
+ "decoder_attention_heads": 12,
13
+ "decoder_ffn_dim": 3072,
14
  "decoder_layerdrop": 0.0,
15
+ "decoder_layers": 6,
16
  "decoder_start_token_id": 2,
17
  "dropout": 0.1,
18
+ "encoder_attention_heads": 12,
19
+ "encoder_ffn_dim": 3072,
20
  "encoder_layerdrop": 0.0,
21
+ "encoder_layers": 6,
22
  "eos_token_id": 2,
23
  "forced_eos_token_id": 2,
24
  "gradient_checkpointing": false,
 
26
  "is_encoder_decoder": true,
27
  "max_position_embeddings": 1024,
28
  "model_type": "mbart",
29
+ "num_hidden_layers": 6,
30
  "pad_token_id": 1,
31
  "scale_embedding": false,
32
  "tokenizer_class": "BartphoTokenizer",
33
  "torch_dtype": "float32",
34
+ "transformers_version": "4.32.1",
35
  "use_cache": true,
36
  "vocab_size": 40030
37
  }
generation_config.json CHANGED
@@ -4,5 +4,5 @@
4
  "eos_token_id": 2,
5
  "forced_eos_token_id": 2,
6
  "pad_token_id": 1,
7
- "transformers_version": "4.31.0"
8
  }
 
4
  "eos_token_id": 2,
5
  "forced_eos_token_id": 2,
6
  "pad_token_id": 1,
7
+ "transformers_version": "4.32.1"
8
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3456d8667ab83840be929e11009726620a0171e2163f44d1e85a29fcd8a23147
3
- size 1583588317
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7f75012bc66f5954194cbcedf19a2cab347f5b941ef83efefbfb579cf006e29
3
+ size 526498157