add marco-o1 and openo1 dataset

2024-11-27 04:20:23 +00:00
parent b26c490ac3
commit 17afb7d410
5 changed files with 38 additions and 2 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -998,6 +998,17 @@ register_model_group(
 )


+register_model_group(
+    models={
+        "Marco-o1-Chat": {
+            DownloadSource.DEFAULT: "AIDC-AI/Marco-o1",
+            DownloadSource.MODELSCOPE: "AIDC-AI/Marco-o1",
+        },
+    },
+    template="marco",
+)
+
+
 register_model_group(
    models={
        "MiniCPM-2B-SFT-Chat": {