finalgp3 commited on
Commit
7c4e5c2
·
verified ·
1 Parent(s): 66cfcc4

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -7
config.json CHANGED
@@ -28,13 +28,14 @@
28
  "summary_use_proj": true,
29
  "task_specific_params": {
30
  "text-generation": {
31
- "early_stopping": true,
32
- "max_new_tokens": 250,
33
- "temperature": 0.5,
34
- "top_p": 0.95,
35
- "do_sample": true,
36
- "repetition_penalty":0.93,
37
- "max_time":120.0
 
38
  }
39
  },
40
  "torch_dtype": "float32",
 
28
  "summary_use_proj": true,
29
  "task_specific_params": {
30
  "text-generation": {
31
+ "max_new_tokens":250,
32
+ "length_penalty":1.0,
33
+ "early_stopping":True,
34
+ "num_return_sequences":1,
35
+ "temperature":0.4,
36
+ "no_repeat_ngram_size":4,
37
+ "top_p":0.80,
38
+ "do_sample":True,
39
  }
40
  },
41
  "torch_dtype": "float32",