add openchat-3.6-8B support

Former-commit-id: b66f39d50d896d7597a1506e67ec210b31c9b700
2024-05-27 20:42:08 +08:00
parent 4bd276f58f
commit 0d9e364a90
2 changed files with 26 additions and 0 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -706,6 +706,15 @@ register_model_group(
    template="openchat",
 )

+register_model_group(
+    models={
+        "OpenChat3.6-8B-Chat": {
+            DownloadSource.DEFAULT: "openchat/openchat-3.6-8b-20240522",
+        }
+    },
+    template="openchat-3.6",
+)
+

 register_model_group(
    models={