add marco-o1 and openo1 dataset

Former-commit-id: 51d49e075470951f109bcdde136203f972450c2e
2025-12-29 10:10:35 +08:00 · 2024-11-27 04:20:23 +00:00
parent 1a8c913d01
commit 7631b97e80
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": {