ColleenMacklin commited on
Commit
1427011
1 Parent(s): 24db40e

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -1
config.json CHANGED
@@ -49,7 +49,18 @@
49
  "summary_use_proj": true,
50
  "torch_dtype": "float32",
51
  "transformers_version": "4.34.1",
52
- "use_cache": true,
 
 
 
 
 
 
 
 
 
 
 
53
  "vocab_size": 50257,
54
  "window_size": 256
55
  }
 
49
  "summary_use_proj": true,
50
  "torch_dtype": "float32",
51
  "transformers_version": "4.34.1",
52
+ "use_cache": true,
53
+ "task_specific_params": {
54
+ "text-generation": {
55
+ "do_sample": true,
56
+ "temperature": 0.75,
57
+ "max_length": 100,
58
+ "top_p": 0.95,
59
+ "top_k": 50,
60
+ "wait_for_model": true,
61
+ "num_return_sequences": 2
62
+ }
63
+ },
64
  "vocab_size": 50257,
65
  "window_size": 256
66
  }