AleBurzio commited on
Commit
2e9c661
1 Parent(s): 8596062

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -2
config.json CHANGED
@@ -28,5 +28,14 @@
28
  "transformers_version": "4.25.0.dev0",
29
  "unk_token_id": 0,
30
  "use_cache": true,
31
- "vocab_size": 250880
32
- }
 
 
 
 
 
 
 
 
 
 
28
  "transformers_version": "4.25.0.dev0",
29
  "unk_token_id": 0,
30
  "use_cache": true,
31
+ "vocab_size": 250880,
32
+ "task_specific_params": {
33
+ "text-generation": {
34
+ "do_sample": true,
35
+ "no_repeat_ngram_size": 2,
36
+ "repetition_penalty": 4.5,
37
+ "max_length": 48
38
+ }
39
+ },
40
+ "word_embed_proj_dim": 512
41
+ }