Skip to content

Commit 953c2a6

Browse files
authored
model : restore support for T5Encoder (#12590)
1 parent d5c6309 commit 953c2a6

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/llama-model.cpp

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11846,10 +11846,11 @@ llm_graph_result_ptr llama_model::build_graph(
1184611846
GGML_ABORT("invalid graph type");
1184711847
};
1184811848
} break;
11849-
//case LLM_ARCH_T5ENCODER:
11850-
// {
11851-
// llm.build_t5_enc(gf);
11852-
// } break;
11849+
case LLM_ARCH_T5ENCODER:
11850+
{
11851+
llm = std::make_unique<llm_build_t5_enc>(*this, params, gf);
11852+
}
11853+
break;
1185311854
case LLM_ARCH_JAIS:
1185411855
{
1185511856
llm = std::make_unique<llm_build_jais>(*this, params, gf);

0 commit comments

Comments
 (0)