marella commited on
Commit
b227990
1 Parent(s): e2e09cd

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +10 -1
config.json CHANGED
@@ -1,3 +1,12 @@
1
  {
2
- "model_type": "gpt2"
 
 
 
 
 
 
 
 
 
3
  }
 
1
  {
2
+ "model_type": "gpt2",
3
+ "task_specific_params": {
4
+ "text-generation": {
5
+ "top_k": 40,
6
+ "top_p": 0.95,
7
+ "temperature": 0.8,
8
+ "repetition_penalty": 1.1,
9
+ "last_n_tokens": 64
10
+ }
11
+ }
12
  }