Use T5ForConditionalGeneration (#7)

- Use T5ForConditionalGeneration (66b9b43093d92186386c748274eb921ed78acaa3) Co-authored-by: Félix Marty <fxmarty@users.noreply.huggingface.co>
2022-12-12 16:10:09 +00:00 · 2022-12-12 16:10:09 +00:00 · 26629ba2cc
parent cb7a9673bc
commit 26629ba2cc
1 changed files with 1 additions and 1 deletions
--- a/config.json
+++ b/config.json
@ -1,6 +1,6 @@
 {
  "architectures": [
-    "T5WithLMHeadModel"
+    "T5ForConditionalGeneration"
  ],
  "d_ff": 4096,
  "d_kv": 64,