Use T5ForConditionalGeneration

#7
by fxmarty - opened
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "T5WithLMHeadModel"
4
  ],
5
  "d_ff": 4096,
6
  "d_kv": 64,
 
1
  {
2
  "architectures": [
3
+ "T5ForConditionalGeneration"
4
  ],
5
  "d_ff": 4096,
6
  "d_kv": 64,