pszemraj commited on
Commit
ea25aeb
1 Parent(s): 6c11e93

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -0
config.json CHANGED
@@ -50,5 +50,13 @@
50
  "transformers_version": "4.25.0.dev0",
51
  "use_cache": true,
52
  "vocab_size": 50257,
 
 
 
 
 
 
 
 
53
  "window_size": 256
54
  }
 
50
  "transformers_version": "4.25.0.dev0",
51
  "use_cache": true,
52
  "vocab_size": 50257,
53
+ "task_specific_params": {
54
+ "text-generation": {
55
+ "do_sample": true,
56
+ "no_repeat_ngram_size": 2,
57
+ "repetition_penalty": 4.5,
58
+ "max_new_tokens": 64
59
+ }
60
+ },
61
  "window_size": 256
62
  }