HenryTbelt commited on
Commit
0bb4d54
·
1 Parent(s): a72c9fc

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -5
config.json CHANGED
@@ -1,7 +1,8 @@
1
  {
2
- "model": "text-generation",
3
- "model_name_or_path": "gpt2",
4
- "config": {},
5
- "tokenizer": {},
6
- "framework": "pt"
 
7
  }
 
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
  }