Merge pull request #5585 from shing100/main

Support EXAONE3.0 Model

Former-commit-id: ef84aaceec87f9fd0f3b5fc0305f1f1e39c1cd60

src/llamafactory/data/template.py

@@ -644,6 +644,14 @@ _register_template(
 )
 
 
+_register_template(
+    name="exaone",
+    format_user=StringFormatter(slots=["[|user|]{{content}}\n[|assistant|]"]),
+    format_system=StringFormatter(slots=["[|system|]{{content}}[|endofturn|]\n"]),
+    format_separator=EmptyFormatter(slots=["\n"]),
+)
+
+
 _register_template(
     name="falcon",
     format_user=StringFormatter(slots=["User: {{content}}\nFalcon:"]),

src/llamafactory/extras/constants.py

@@ -472,6 +472,16 @@ register_model_group(
 )
 
 
+register_model_group(
+    models={
+        "EXAONE-3.0-7.8B-Instruct": {
+            DownloadSource.DEFAULT: "LGAI-EXAONE/EXAONE-3.0-7.8B-Instruct",
+        },
+    },
+    template="exaone",
+)
+
+
 register_model_group(
     models={
         "Falcon-7B": {