ColleenMacklin commited on
Commit
b78d445
1 Parent(s): 0f8a8a1

Update config.json

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