From a4986abc64c580966eb8dd0ed61ceafaeb0d1d45 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <sgugger@users.noreply.huggingface.co> Date: Mon, 12 Dec 2022 16:09:56 +0000 Subject: [PATCH] Use T5ForConditionalGeneration (#8) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - Use T5ForConditionalGeneration (6421ace5cab13343b167e259b63b7608bb2a40bf) Co-authored-by: Félix Marty <fxmarty@users.noreply.huggingface.co> --- config.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.json b/config.json index da65c9e..c360f5f 100644 --- a/config.json +++ b/config.json @@ -1,6 +1,6 @@ { "architectures": [ - "T5WithLMHeadModel" + "T5ForConditionalGeneration" ], "d_ff": 3072, "d_kv": 64,