pszemraj commited on
Commit
18099b9
1 Parent(s): 231d6af

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -0
config.json CHANGED
@@ -25,5 +25,13 @@
25
  "transformers_version": "4.25.0.dev0",
26
  "use_cache": true,
27
  "vocab_size": 50272,
 
 
 
 
 
 
 
 
28
  "word_embed_proj_dim": 512
29
  }
 
25
  "transformers_version": "4.25.0.dev0",
26
  "use_cache": true,
27
  "vocab_size": 50272,
28
+ "task_specific_params": {
29
+ "text-generation": {
30
+ "do_sample": true,
31
+ "no_repeat_ngram_size": 2,
32
+ "repetition_penalty": 4.5,
33
+ "max_new_tokens": 64
34
+ }
35
+ },
36
  "word_embed_proj_dim": 512
37
  }