HenryTbelt commited on
Commit
40cb3f7
1 Parent(s): 1512cbb

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -3
config.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "model_type": "t5",
3
- "model_name_or_path": "t5-base",
4
  "max_length": 128,
5
- "temperature": 0.1,
6
  "repetition_penalty": 1.0,
7
  "num_beams": 4
8
  }
 
1
  {
2
+ "model_type": "text2text_generation",
3
+ "model_name_or_path": "google/pegasus-xsum",
4
  "max_length": 128,
5
+ "temperature": 0.7,
6
  "repetition_penalty": 1.0,
7
  "num_beams": 4
8
  }