MRNH commited on
Commit
5e7100e
1 Parent(s): 4df752a

Upload MBartForConditionalGeneration

Browse files
Files changed (3) hide show
  1. config.json +1 -0
  2. generation_config.json +13 -0
  3. pytorch_model.bin +3 -0
config.json CHANGED
@@ -52,6 +52,7 @@
52
  "scale_embedding": true,
53
  "static_position_embeddings": false,
54
  "tokenizer_class": "MBart50Tokenizer",
 
55
  "transformers_version": "4.31.0",
56
  "use_cache": true,
57
  "vocab_size": 250057
 
52
  "scale_embedding": true,
53
  "static_position_embeddings": false,
54
  "tokenizer_class": "MBart50Tokenizer",
55
+ "torch_dtype": "float32",
56
  "transformers_version": "4.31.0",
57
  "use_cache": true,
58
  "vocab_size": 250057
generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "decoder_start_token_id": 2,
5
+ "early_stopping": true,
6
+ "eos_token_id": 2,
7
+ "forced_eos_token_id": 2,
8
+ "max_length": 200,
9
+ "num_beams": 5,
10
+ "output_hidden_states": true,
11
+ "pad_token_id": 1,
12
+ "transformers_version": "4.31.0"
13
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d171f054f25357c68c23852bf9ca0b1dfb2d5992d265fd36c277a398568528fb
3
+ size 2444706333