fix

ztxz16 · Sep 22, 2024 · 0222fa1 · 0222fa1
1 parent 72e5cba
commit 0222fa1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/model.cpp b/src/model.cpp
@@ -16,6 +16,7 @@
 #include "minicpm.h"
 #include "internlm2.h"
 #include "bert.h"
+#include "xlmroberta.h"
 #include "graphllm.h"
 #include "phi3.h"
 
@@ -194,6 +195,8 @@ namespace fastllm {
             model = (basellm*)(new GLMModel());
         } else if (modelType == "bert") {
             model = (basellm*)(new BertModel());
+        } else if (modelType == "xlm-roberta") {
+            model = (basellm*)(new XlmRobertaModel());
         } else if (modelType == "fastllmJson") {
             model = new GraphLLMModel("fastllmJson");
         } else {