X-Wang commited on
Commit
4bbb896
·
1 Parent(s): 22b6ff3

X-Wang/pruned-mt5-small

Browse files
Files changed (2) hide show
  1. generation_config.json +3 -0
  2. pytorch_model.bin +1 -1
generation_config.json CHANGED
@@ -1,6 +1,9 @@
1
  {
2
  "decoder_start_token_id": 0,
 
3
  "eos_token_id": 1,
 
 
4
  "pad_token_id": 0,
5
  "transformers_version": "4.31.0"
6
  }
 
1
  {
2
  "decoder_start_token_id": 0,
3
+ "do_sample": true,
4
  "eos_token_id": 1,
5
+ "max_length": 512,
6
+ "num_beams": 6,
7
  "pad_token_id": 0,
8
  "transformers_version": "4.31.0"
9
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0733bcf8bcb3f0e8ecd8323bc98fa564099caf3078cafd9e0f321f971cd2be18
3
  size 342099461
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ce4e0efec3f9eac0dce1d532cf0c9a06d39d53a42aa347008a04b4074f65610
3
  size 342099461