sgugger fxmarty commited on
Commit
a4986ab
1 Parent(s): 23aa4f4

Use T5ForConditionalGeneration (#8)

Browse files

- Use T5ForConditionalGeneration (6421ace5cab13343b167e259b63b7608bb2a40bf)


Co-authored-by: Félix Marty <fxmarty@users.noreply.huggingface.co>

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": 3072,
6
  "d_kv": 64,
 
1
  {
2
  "architectures": [
3
+ "T5ForConditionalGeneration"
4
  ],
5
  "d_ff": 3072,
6
  "d_kv": 64,