Fix architecutre (#7)

- Fix architecutre (91ab3295459e5fcff3b6472bce4a3bb2a144b989) Co-authored-by: Omar Sanseviero <osanseviero@users.noreply.huggingface.co>
2022-12-12 16:06:17 +00:00 · 2022-12-12 16:06:17 +00:00 · c2bea447bc
parent d78aea13fa
commit c2bea447bc
1 changed files with 1 additions and 1 deletions
--- a/config.json
+++ b/config.json
@ -1,6 +1,6 @@
 {
  "architectures": [
-    "T5WithLMHeadModel"
+    "T5ForConditionalGeneration",
  ],
  "d_ff": 2048,
  "d_kv": 64,