Ver Fonte

model : restore support for T5Encoder (#12590)

HighDoping há 9 meses atrás
pai
commit
953c2a62cf
1 ficheiros alterados com 5 adições e 4 exclusões
  1. 5 4
      src/llama-model.cpp

+ 5 - 4
src/llama-model.cpp

@@ -11846,10 +11846,11 @@ llm_graph_result_ptr llama_model::build_graph(
                         GGML_ABORT("invalid graph type");
                         GGML_ABORT("invalid graph type");
                 };
                 };
             } break;
             } break;
-            //case LLM_ARCH_T5ENCODER:
-            //    {
-            //        llm.build_t5_enc(gf);
-            //    } break;
+        case LLM_ARCH_T5ENCODER:
+            {
+                llm = std::make_unique<llm_build_t5_enc>(*this, params, gf);
+            }
+            break;
         case LLM_ARCH_JAIS:
         case LLM_ARCH_JAIS:
             {
             {
                 llm = std::make_unique<llm_build_jais>(*this, params, gf);
                 llm = std::make_unique<llm_build_jais>(*this, params, gf);