From c2bea447bc4e723126e9c5f126a9449c698168e1 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Mon, 12 Dec 2022 16:06:17 +0000 Subject: [PATCH] Fix architecutre (#7) - Fix architecutre (91ab3295459e5fcff3b6472bce4a3bb2a144b989) Co-authored-by: Omar Sanseviero --- config.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.json b/config.json index 535be25..6c784f9 100644 --- a/config.json +++ b/config.json @@ -1,6 +1,6 @@ { "architectures": [ - "T5WithLMHeadModel" + "T5ForConditionalGeneration", ], "d_ff": 2048, "d_kv": 64,