nakcnx commited on
Commit
096af66
1 Parent(s): 4875540

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -0
config.json CHANGED
@@ -46,6 +46,17 @@
46
  "summary_proj_to_labels": true,
47
  "summary_type": "cls_index",
48
  "summary_use_proj": true,
 
 
 
 
 
 
 
 
 
 
 
49
  "torch_dtype": "float32",
50
  "transformers_version": "4.26.0",
51
  "use_cache": true,
 
46
  "summary_proj_to_labels": true,
47
  "summary_type": "cls_index",
48
  "summary_use_proj": true,
49
+ "task_specific_params": {
50
+ "text-generation": {
51
+ "do_sample": true,
52
+ "max_length": 200,
53
+ "num_beams": 3,
54
+ "temperature": 1,
55
+ "no_repeat_ngram_size": 2,
56
+ "top_k": 50,
57
+ "top_p": 80
58
+ }
59
+ },
60
  "torch_dtype": "float32",
61
  "transformers_version": "4.26.0",
62
  "use_cache": true,