Skip to content

Commit f7c0b91

Browse files
authored
Fix T5ENCODER model handling.
1 parent 5ed38b6 commit f7c0b91

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)