Use T5ForConditionalGeneration (#8)

- Use T5ForConditionalGeneration (6421ace5cab13343b167e259b63b7608bb2a40bf) Co-authored-by: Félix Marty <fxmarty@users.noreply.huggingface.co>
2022-12-12 16:09:56 +00:00 · 2022-12-12 16:09:56 +00:00 · a4986abc64
parent 23aa4f41cb
commit a4986abc64
1 changed files with 1 additions and 1 deletions
--- a/config.json
+++ b/config.json
@ -1,6 +1,6 @@
 {
  "architectures": [
-    "T5WithLMHeadModel"
+    "T5ForConditionalGeneration"
  ],
  "d_ff": 3072,
  "d_kv": 64,