MRNH commited on
Commit
3f926fa
1 Parent(s): f7d2156

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:16dec13a38d88e6d8e9218a19d7e9fed504e3c762a80149a3d5c73b1d249f77e
3
+ size 2444706333