Merge pull request #3921 from gusye1234/main

Add openchat-3.6-8B support Former-commit-id: 92e6bba3cab22b7835a68f787caf7992a398978e
2024-05-27 20:52:37 +08:00
parent 3b28c003dd 8c9b929bb0
commit a886544d3d
2 changed files with 25 additions and 0 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -719,6 +719,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={